結果
問題 |
No.2360 Path to Integer
|
ユーザー |
|
提出日時 | 2024-09-11 11:49:38 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 162 ms / 2,500 ms |
コード長 | 2,115 bytes |
コンパイル時間 | 3,953 ms |
コンパイル使用メモリ | 290,376 KB |
実行使用メモリ | 21,684 KB |
最終ジャッジ日時 | 2024-09-11 11:49:45 |
合計ジャッジ時間 | 6,572 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 15 |
ソースコード
#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> bfs; queue<int> Q; Q.push(0); while (!Q.empty()) { int v = Q.front(); Q.pop(); bfs.push_back(v); for (int w : E[v]) { if (w != p[v]) { p[w] = v; c[v].push_back(w); Q.push(w); } } } reverse(bfs.begin(), bfs.end()); vector<int> dp1(N, 0);//部分木のサイズ vector<long long> dp2(N, 0); //C[i] = stoll(A[i]) //B[i] = 10^(A[i].size()) % MOD for (int v : bfs) { dp1[v] = 1; for (int w : c[v]) { dp2[v] += dp2[w]; dp2[v] %= MOD; dp1[v] += dp1[w]; } dp2[v] *= B[v]; dp2[v] %= MOD; dp2[v] += dp1[v] * C[v] % MOD;//vは部分木のすべての頂点の数足される。 dp2[v] %= MOD; } reverse(bfs.begin(), bfs.end()); vector<long long> dp3(N); dp3[0] = dp2[0]; for (int v : bfs) { //vを根としたものは求まっている。 long long sum = dp3[v]; for (int w : c[v]) { long long sum2 = sum - dp2[w] * B[v] % MOD + MOD;//wの部分木を切り離す(v -> wの辺削除) sum2 %= MOD; sum2 = sum2 - C[v] * dp1[w] % MOD + MOD;//wの部分木に寄与していたC[v]を削除 sum2 %= MOD; //w <- vに辺を追加 sum2 *= B[w]; sum2 %= MOD; sum2 += C[w] * (N - dp1[w]);//wについてはdp2[w]に含まれているため除く。 sum2 %= MOD; dp3[w] = dp2[w] + sum2; } } long long ans = 0; for (int i = 0; i < N; i++) { ans += dp3[i]; ans %= MOD; } cout << ans << endl; }