結果

問題 No.2949 Product on Tree
コンテスト
ユーザー askr58
提出日時 2024-10-25 22:16:35
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 317 ms / 2,000 ms
コード長 743 bytes
コンパイル時間 5,221 ms
コンパイル使用メモリ 312,580 KB
実行使用メモリ 22,020 KB
最終ジャッジ日時 2024-10-25 22:16:57
合計ジャッジ時間 21,001 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using mint=modint998244353;
int main()
{
    int n;
    cin>>n;
    vector<ll> a(n);
    for(int i=0;i<n;i++)cin>>a[i];
    vector<vector<int>> graph(n);
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        u--;v--;
        graph[u].push_back(v);
        graph[v].push_back(u);
    }
    mint ans=0;
    auto dfs=[&](auto dfs,int v,int par)->mint{
        mint res=a[v],t=0;
        for(int u:graph[v]){
            if(u==par)continue;
            t=dfs(dfs,u,v);
            ans+=res*t;
            res+=t*a[v];
        }
        return res;
    };
    dfs(dfs,0,-1);
    cout<<ans.val()<<endl;
    return 0;
}
0