結果
問題 | 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 setrecursionlimitsetrecursionlimit(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 = 0def dfs(v,p):v0 = a[v]>>i&1^1; v1 = a[v]>>i&1for c in adj[v]:if c==p: continuec0,c1 = dfs(c,v)v0,v1 = (v0*c0+v1*c1+v0*c1)%mod,(v0*c1+v1*c0+v1*c1)%modreturn v0,v1for i in range(30): ans += (1<<i)*dfs(0,-1)[1]print(ans%mod)