結果
問題 |
No.2360 Path to Integer
|
ユーザー |
|
提出日時 | 2024-09-11 11:53:41 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 151 ms / 2,500 ms |
コード長 | 1,926 bytes |
コンパイル時間 | 3,942 ms |
コンパイル使用メモリ | 291,516 KB |
実行使用メモリ | 20,132 KB |
最終ジャッジ日時 | 2024-09-11 11:53:47 |
合計ジャッジ時間 | 6,195 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 15 |
ソースコード
#include<bits/stdc++.h> #include<atcoder/modint> using namespace std; using namespace atcoder; using mint = modint998244353; 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<mint> 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; } C[i] = stoll(A[i]); } 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<mint> 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]; dp1[v] += dp1[w]; } dp2[v] *= B[v]; dp2[v] += dp1[v] * C[v];//vは部分木のすべての頂点の数足される。 } reverse(bfs.begin(), bfs.end()); vector<mint> dp3(N); dp3[0] = dp2[0]; for (int v : bfs) { //vを根としたものは求まっている。 mint sum = dp3[v]; for (int w : c[v]) { mint sum2 = sum - dp2[w] * B[v];//wの部分木を切り離す(v -> wの辺削除) sum2 -= C[v] * dp1[w];//wの部分木に寄与していたC[v]を削除 //w <- vに辺を追加 sum2 *= B[w]; sum2 += C[w] * (N - dp1[w]);//wについてはdp2[w]に含まれているため除く。 dp3[w] = dp2[w] + sum2; } } mint ans = 0; for (int i = 0; i < N; i++) { ans += dp3[i]; } cout << ans.val() << endl; }