結果
問題 | No.2360 Path to Integer |
ユーザー | SSRS |
提出日時 | 2023-06-23 21:52:18 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 141 ms / 2,500 ms |
コード長 | 1,854 bytes |
コンパイル時間 | 2,550 ms |
コンパイル使用メモリ | 180,236 KB |
実行使用メモリ | 21,620 KB |
最終ジャッジ日時 | 2023-09-13 16:52:49 |
合計ジャッジ時間 | 3,867 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 2 ms
4,376 KB |
testcase_06 | AC | 2 ms
4,376 KB |
testcase_07 | AC | 2 ms
4,376 KB |
testcase_08 | AC | 13 ms
4,660 KB |
testcase_09 | AC | 141 ms
20,380 KB |
testcase_10 | AC | 108 ms
19,680 KB |
testcase_11 | AC | 113 ms
19,776 KB |
testcase_12 | AC | 117 ms
20,060 KB |
testcase_13 | AC | 140 ms
20,168 KB |
testcase_14 | AC | 141 ms
21,620 KB |
testcase_15 | AC | 132 ms
20,168 KB |
testcase_16 | AC | 128 ms
21,600 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; int main(){ int N; cin >> N; vector<string> A(N); for (int i = 0; i < N; i++){ cin >> A[i]; } vector<vector<int>> E(N); for (int i = 0; i < N - 1; i++){ int u, v; cin >> u >> v; u--; v--; E[u].push_back(v); E[v].push_back(u); } vector<long long> B(N), C(N); for (int i = 0; i < N; i++){ B[i] = 1; for (int j = 0; j < A[i].size(); j++){ B[i] *= 10; B[i] %= MOD; } C[i] = stoll(A[i]) % MOD; } vector<int> p(N, -1); vector<vector<int>> c(N); vector<int> b; queue<int> Q; Q.push(0); while (!Q.empty()){ int v = Q.front(); Q.pop(); b.push_back(v); for (int w : E[v]){ if (w != p[v]){ p[w] = v; c[v].push_back(w); Q.push(w); } } } reverse(b.begin(), b.end()); vector<int> dp1(N, 0); vector<long long> dp2(N, 0); for (int v : b){ dp1[v] = 1; for (int w : c[v]){ dp1[v] += dp1[w]; dp2[v] += dp2[w]; dp2[v] %= MOD; } dp1[v] %= MOD; dp2[v] *= B[v]; dp2[v] %= MOD; dp2[v] += C[v] * dp1[v]; dp2[v] %= MOD; } reverse(b.begin(), b.end()); vector<long long> dp3(N, 0); dp3[0] = C[0]; for (int v : b){ long long sum = 0; for (int w : c[v]){ sum += dp2[w]; } sum %= MOD; for (int w : c[v]){ long long sum2 = (sum - dp2[w] + MOD) % MOD; sum2 *= B[v]; sum2 %= MOD; sum2 += C[v] * (dp1[v] - dp1[w] - 1); sum2 %= MOD; sum2 += dp3[v]; sum2 %= MOD; sum2 *= B[w]; sum2 %= MOD; sum2 += C[w] * (N - dp1[w] + 1); sum2 %= MOD; dp3[w] = sum2; } } long long ans = 0; for (int i = 0; i < N; i++){ ans += dp2[i] + dp3[i] - C[i] + MOD; ans %= MOD; } cout << ans << endl; }