結果

問題 No.2949 Product on Tree
ユーザー nouka28nouka28
提出日時 2024-09-23 07:29:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,679 ms / 2,000 ms
コード長 510 bytes
コンパイル時間 321 ms
コンパイル使用メモリ 81,812 KB
実行使用メモリ 370,792 KB
最終ジャッジ日時 2024-09-23 07:30:27
合計ジャッジ時間 40,171 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,764 KB
testcase_01 AC 36 ms
51,940 KB
testcase_02 AC 37 ms
52,140 KB
testcase_03 AC 489 ms
111,316 KB
testcase_04 AC 526 ms
108,696 KB
testcase_05 AC 499 ms
110,780 KB
testcase_06 AC 550 ms
111,764 KB
testcase_07 AC 499 ms
108,832 KB
testcase_08 AC 555 ms
112,012 KB
testcase_09 AC 523 ms
113,000 KB
testcase_10 AC 552 ms
119,056 KB
testcase_11 AC 734 ms
140,348 KB
testcase_12 AC 580 ms
136,228 KB
testcase_13 AC 810 ms
176,728 KB
testcase_14 AC 1,064 ms
247,096 KB
testcase_15 AC 1,152 ms
244,172 KB
testcase_16 AC 1,094 ms
237,264 KB
testcase_17 AC 827 ms
206,468 KB
testcase_18 AC 1,211 ms
264,540 KB
testcase_19 AC 1,459 ms
315,704 KB
testcase_20 AC 1,136 ms
237,012 KB
testcase_21 AC 1,063 ms
221,916 KB
testcase_22 AC 1,494 ms
335,776 KB
testcase_23 AC 574 ms
111,572 KB
testcase_24 AC 557 ms
112,044 KB
testcase_25 AC 559 ms
111,824 KB
testcase_26 AC 554 ms
111,432 KB
testcase_27 AC 560 ms
112,164 KB
testcase_28 AC 558 ms
111,968 KB
testcase_29 AC 598 ms
113,068 KB
testcase_30 AC 573 ms
116,592 KB
testcase_31 AC 625 ms
123,356 KB
testcase_32 AC 916 ms
164,812 KB
testcase_33 AC 1,131 ms
231,464 KB
testcase_34 AC 1,411 ms
337,196 KB
testcase_35 AC 1,224 ms
248,696 KB
testcase_36 AC 1,468 ms
364,624 KB
testcase_37 AC 1,156 ms
293,796 KB
testcase_38 AC 1,057 ms
260,832 KB
testcase_39 AC 1,342 ms
307,384 KB
testcase_40 AC 1,679 ms
370,792 KB
testcase_41 AC 932 ms
239,092 KB
testcase_42 AC 817 ms
208,992 KB
testcase_43 AC 215 ms
104,580 KB
testcase_44 AC 217 ms
104,580 KB
testcase_45 AC 263 ms
117,056 KB
testcase_46 AC 238 ms
106,528 KB
testcase_47 AC 194 ms
102,184 KB
testcase_48 AC 243 ms
106,280 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(2*10**5)

mod=998244353

n=int(input())

a=list(map(int,input().split()))

g=[[]for i in range(n)]

for i in range(n-1):
	u,v=map(int,input().split())
	u-=1
	v-=1
	g[u].append(v)
	g[v].append(u)

ans=0
inv2=pow(2,mod-2,mod)
def dfs(p,prev):
	global ans
	
	sm1=0
	sm2=0

	for e in g[p]:
		if e==prev:continue
		v=dfs(e,p)
		sm1=(sm1+v)%mod
		sm2=(sm2+v*v)%mod
	
	ans=(ans+a[p]*sm1)%mod

	ans=(ans+a[p]*(sm1*sm1-sm2)%mod*inv2)%mod

	return a[p]*(1+sm1)%mod

dfs(0,-1)

print(ans)
0