結果
問題 | No.2377 SUM AND XOR on Tree |
ユーザー |
![]() |
提出日時 | 2023-07-07 23:15:05 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 3,905 ms / 4,000 ms |
コード長 | 533 bytes |
コンパイル時間 | 319 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 53,004 KB |
最終ジャッジ日時 | 2024-07-21 19:39:40 |
合計ジャッジ時間 | 6,883 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
from sys import setrecursionlimit setrecursionlimit(10**7) n = int(input()) adj = [[] for _ in range(n)] for _ in range(n-1): u,v = map(int,input().split()) adj[u-1].append(v-1); adj[v-1].append(u-1) a = list(map(int,input().split())) mod = 998244353; ans = 0 def dfs(v,p): v0 = a[v]>>i&1^1; v1 = a[v]>>i&1 for c in adj[v]: if c==p: continue c0,c1 = dfs(c,v) v0,v1 = (v0*c0+v1*c1+v0*c1)%mod,(v0*c1+v1*c0+v1*c1)%mod return v0,v1 for i in range(30): ans += (1<<i)*dfs(0,-1)[1] print(ans%mod)