結果

問題 No.2949 Product on Tree
ユーザー askr58askr58
提出日時 2024-10-25 22:16:35
言語 C++23
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 1 ms
6,820 KB
testcase_03 AC 262 ms
15,584 KB
testcase_04 AC 272 ms
15,304 KB
testcase_05 AC 302 ms
15,544 KB
testcase_06 AC 295 ms
15,624 KB
testcase_07 AC 281 ms
15,420 KB
testcase_08 AC 274 ms
15,664 KB
testcase_09 AC 279 ms
15,972 KB
testcase_10 AC 277 ms
16,000 KB
testcase_11 AC 287 ms
16,440 KB
testcase_12 AC 297 ms
17,212 KB
testcase_13 AC 263 ms
17,348 KB
testcase_14 AC 301 ms
18,016 KB
testcase_15 AC 277 ms
19,088 KB
testcase_16 AC 260 ms
18,204 KB
testcase_17 AC 270 ms
18,736 KB
testcase_18 AC 269 ms
18,576 KB
testcase_19 AC 309 ms
19,660 KB
testcase_20 AC 303 ms
19,348 KB
testcase_21 AC 307 ms
19,548 KB
testcase_22 AC 286 ms
19,860 KB
testcase_23 AC 281 ms
15,928 KB
testcase_24 AC 279 ms
15,892 KB
testcase_25 AC 303 ms
15,748 KB
testcase_26 AC 284 ms
15,748 KB
testcase_27 AC 291 ms
15,992 KB
testcase_28 AC 285 ms
15,984 KB
testcase_29 AC 274 ms
16,000 KB
testcase_30 AC 296 ms
16,176 KB
testcase_31 AC 280 ms
16,876 KB
testcase_32 AC 311 ms
16,856 KB
testcase_33 AC 282 ms
18,432 KB
testcase_34 AC 294 ms
20,348 KB
testcase_35 AC 297 ms
19,016 KB
testcase_36 AC 285 ms
20,996 KB
testcase_37 AC 317 ms
21,184 KB
testcase_38 AC 316 ms
19,980 KB
testcase_39 AC 291 ms
20,224 KB
testcase_40 AC 307 ms
21,848 KB
testcase_41 AC 286 ms
20,800 KB
testcase_42 AC 299 ms
22,020 KB
testcase_43 AC 135 ms
13,324 KB
testcase_44 AC 139 ms
13,280 KB
testcase_45 AC 183 ms
16,300 KB
testcase_46 AC 167 ms
15,020 KB
testcase_47 AC 120 ms
11,884 KB
testcase_48 AC 166 ms
15,188 KB
権限があれば一括ダウンロードができます

ソースコード

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