結果
問題 | No.1420 国勢調査 (Easy) |
ユーザー |
![]() |
提出日時 | 2021-03-06 11:34:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 848 ms / 2,000 ms |
コード長 | 1,094 bytes |
コンパイル時間 | 95 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 59,500 KB |
最終ジャッジ日時 | 2024-10-08 06:49:33 |
合計ジャッジ時間 | 18,088 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
from sys import setrecursionlimit, stdin from collections import deque def find(parent, i): t = parent[i] if t < 0: return i t = find(parent, t) parent[i] = t return t def unite(parent, i, j): i = find(parent, i) j = find(parent, j) if i == j: return parent[j] += parent[i] parent[i] = j readline = stdin.readline setrecursionlimit(10 ** 6) N, M = map(int, readline().split()) parent = [-1] * N d = {} vs = [] for _ in range(M): A, B = map(lambda x: int(x) - 1, readline().split()) Y = int(readline()) vs.append((A, B, Y)) d.setdefault(A, {}) d.setdefault(B, {}) d[A][B] = Y d[B][A] = Y unite(parent, A, B) result = [-1] * N q = deque(i for i in range(N) if parent[i] < 0) for i in q: result[i] = 0 while q: a = q.popleft() d.setdefault(a, {}) for b in d[a]: if result[b] != -1: continue result[b] = result[a] ^ d[a][b] q.append(b) for A, B, Y in vs: if result[A] ^ result[B] != Y: print(-1) exit() print(*result, sep='\n')