結果
問題 | No.2949 Product on Tree |
ユーザー |
![]() |
提出日時 | 2024-10-26 14:06:28 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 621 bytes |
コンパイル時間 | 465 ms |
コンパイル使用メモリ | 82,784 KB |
実行使用メモリ | 467,056 KB |
最終ジャッジ日時 | 2024-10-26 14:07:19 |
合計ジャッジ時間 | 48,537 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 45 TLE * 1 |
ソースコード
import syssys.setrecursionlimit(200050)MOD = 998244353N = int(input())A = list(map(int, input().split()))UV = [list(map(int, input().split())) for _ in range(N - 1)]E = [[] for _ in range(N)]for u, v in UV:u -= 1v -= 1E[u].append(v)E[v].append(u)def dfs(x, p):global ans, Xfor y in E[x]:if y != p:X[y] += (X[x] + Y[x] + A[x]) * A[y]X[y] %= MODdfs(y, x)Y[x] += (Y[y]+ A[y]) * A[x]Y[x] %= MOD#print(x,X,Y)X = [0] * NY = [0] * Nans = 0dfs(0, -1)#print(X,sum(X))#print(Y,sum(Y))print(sum(X) % MOD)