結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
9,532 KB
testcase_01 AC 5 ms
10,268 KB
testcase_02 AC 5 ms
9,480 KB
testcase_03 AC 159 ms
16,428 KB
testcase_04 AC 152 ms
16,424 KB
testcase_05 AC 161 ms
16,572 KB
testcase_06 AC 160 ms
16,564 KB
testcase_07 AC 154 ms
16,296 KB
testcase_08 AC 162 ms
16,608 KB
testcase_09 AC 176 ms
16,932 KB
testcase_10 AC 170 ms
17,068 KB
testcase_11 AC 166 ms
17,416 KB
testcase_12 AC 173 ms
18,804 KB
testcase_13 AC 165 ms
19,292 KB
testcase_14 AC 149 ms
20,024 KB
testcase_15 AC 158 ms
21,304 KB
testcase_16 AC 174 ms
20,300 KB
testcase_17 AC 164 ms
20,600 KB
testcase_18 AC 167 ms
20,548 KB
testcase_19 AC 170 ms
21,696 KB
testcase_20 AC 161 ms
21,504 KB
testcase_21 AC 179 ms
21,512 KB
testcase_22 AC 168 ms
22,224 KB
testcase_23 AC 165 ms
16,660 KB
testcase_24 AC 163 ms
16,568 KB
testcase_25 AC 154 ms
16,592 KB
testcase_26 AC 187 ms
16,572 KB
testcase_27 AC 162 ms
16,644 KB
testcase_28 AC 159 ms
16,684 KB
testcase_29 AC 154 ms
16,836 KB
testcase_30 AC 165 ms
17,160 KB
testcase_31 AC 171 ms
17,968 KB
testcase_32 AC 176 ms
18,036 KB
testcase_33 AC 173 ms
20,012 KB
testcase_34 AC 177 ms
22,460 KB
testcase_35 AC 175 ms
20,924 KB
testcase_36 AC 198 ms
23,328 KB
testcase_37 AC 186 ms
23,740 KB
testcase_38 AC 179 ms
21,896 KB
testcase_39 AC 181 ms
22,212 KB
testcase_40 AC 176 ms
24,568 KB
testcase_41 AC 170 ms
23,012 KB
testcase_42 AC 183 ms
24,772 KB
testcase_43 AC 58 ms
15,148 KB
testcase_44 AC 59 ms
15,164 KB
testcase_45 AC 74 ms
16,868 KB
testcase_46 AC 67 ms
16,144 KB
testcase_47 AC 49 ms
14,428 KB
testcase_48 AC 69 ms
16,300 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
}
0