結果
問題 |
No.2360 Path to Integer
|
ユーザー |
|
提出日時 | 2024-09-12 11:13:23 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 151 ms / 2,500 ms |
コード長 | 1,651 bytes |
コンパイル時間 | 4,424 ms |
コンパイル使用メモリ | 290,744 KB |
実行使用メモリ | 20,132 KB |
最終ジャッジ日時 | 2024-09-12 11:13:31 |
合計ジャッジ時間 | 7,122 ms |
ジャッジサーバーID (参考情報) |
judge5 / 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<mint> B(N), C(N); for (int i = 0; i < N; i++) { B[i] = 1; for (int j = 0; j < (int) A[i].size(); j++) { B[i] *= 10; } C[i] = mint(stoll(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<vector<int>> c(N); vector<int> p(N, -1); 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); vector<mint> dp2(N); for (int v : bfs) { dp1[v]++; for (int w : c[v]) { dp1[v] += dp1[w]; dp2[v] += dp2[w]; } dp2[v] *= B[v]; dp2[v] += C[v] * dp1[v]; } reverse(bfs.begin(), bfs.end()); vector<mint> dp3(N);//dp3[0] = 0; for (int v : bfs) { mint sum = 0; for (int w : c[v]) { sum += dp2[w]; } for (int w : c[v]) { mint sum2 = sum; sum2 -= dp2[w]; sum2 *= B[v]; sum2 += C[v] * (dp1[v] - dp1[w]); sum2 += dp3[v]; sum2 *= B[w]; sum2 += C[w] * (N - dp1[w]); dp3[w] = sum2; } } mint ans = 0; for (int i = 0; i < N; i++) { ans += dp2[i] + dp3[i]; } cout << ans.val() << endl; }