結果

問題 No.2726 Rooted Tree Nim
コンテスト
ユーザー PNJ
提出日時 2024-04-12 22:34:35
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 774 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 154,124 KB
最終ジャッジ日時 2024-10-02 23:31:54
合計ジャッジ時間 6,814 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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
  visit = []
  while len(todo):
    s = todo.pop()
    visit.append(s)
    for t in G[s]:
      if dist[t] == -1:
        parent[t] = s
        dist[t] = dist[s] + 1
        todo.append(t)
  for s in range(N):
    if dist[s] % 2 == 0:
      A[s] = 0
  while len(visit):
    s = visit.pop()
    if s != 0:
      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