結果
問題 | No.2949 Product on Tree |
ユーザー |
|
提出日時 | 2024-10-25 21:35:22 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 198 ms / 2,000 ms |
コード長 | 565 bytes |
コンパイル時間 | 808 ms |
コンパイル使用メモリ | 73,836 KB |
実行使用メモリ | 24,772 KB |
最終ジャッジ日時 | 2024-10-25 21:35:40 |
合計ジャッジ時間 | 10,901 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
#include<iostream> #include<vector> #include<cassert> #include<atcoder/modint> using namespace std; using mint=atcoder::modint998244353; int N; int A[2<<17]; vector<int>G[2<<17]; mint ans; mint dfs(int u,int p) { mint cur=A[u]; for(int v:G[u])if(v!=p) { mint t=dfs(v,u); ans+=t*cur; cur+=t*A[u]; } return cur; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cin>>N; for(int i=0;i<N;i++)cin>>A[i]; for(int i=0;i<N-1;i++) { int u,v;cin>>u>>v;u--,v--; G[u].push_back(v); G[v].push_back(u); } dfs(0,-1); cout<<ans.val()<<endl; }