結果

問題 No.2726 Rooted Tree Nim
コンテスト
ユーザー PNJ
提出日時 2024-04-12 22:30:54
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 722 bytes
コンパイル時間 180 ms
コンパイル使用メモリ 82,148 KB
実行使用メモリ 156,836 KB
最終ジャッジ日時 2024-10-02 23:30:08
合計ジャッジ時間 9,018 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 5 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

T = int(input())
for _ in range(T):
  N,K = 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)
  A = list(map(int,input().split()))
  parent = [-1 for i in range(N)]
  todo = [0]
  dist = [-1 for i in range(N)]
  dist[0] = 0
  while len(todo):
    s = todo.pop()
    for t in G[s]:
      if dist[t] == -1:
        parent[t] = s
        dist[t] = dist[s] + 1
        todo.append(t)
  H = [(dist[s],s) for s in range(N)]
  H.sort()
  while len(H):
    d,s = H.pop()
    if d % 2:
      A[parent[s]] += A[s]
  g = 0
  for s in G[0]:
    g ^= (A[s] % (K + 1))
  if g:
    print('K')
  else:
    print('P')
0