結果

問題 No.1420 国勢調査 (Easy)
ユーザー LyricalMaestro
提出日時 2025-04-19 13:35:52
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,358 bytes
コンパイル時間 793 ms
コンパイル使用メモリ 82,480 KB
実行使用メモリ 294,304 KB
最終ジャッジ日時 2025-04-19 13:36:22
合計ジャッジ時間 21,391 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 25 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

## https://yukicoder.me/problems/no/1420

from collections import deque


def solve(N, edges, k):
    next_nodes  =[[] for _ in range(N)]
    for i, j, y  in edges:
        y0 = 1 if (y & (1 << k) > 0) else 0
        next_nodes[i].append((j, y0))
        next_nodes[j].append((i, y0))

    queue = deque()    
    values = [-1] * N
    for s_i in range(N):
        if values[s_i] == -1:
            values[s_i] = 0
            queue.append(s_i)
            while len(queue) > 0:
                v = queue.popleft()
                for w, y in next_nodes[v]:
                    x = y ^ values[v]
                    if values[w] == -1:
                        values[w] = x
                        queue.append(w)
                    elif values[w] != x:
                        return -1

    return values



def main():
    N, M = map(int, input().split())
    edges = []
    for _ in range(M):
        a, b = map(int, input().split())
        y = int(input())
        edges.append((a - 1, b - 1, y))
    
    answers = [0] * N
    for k in range(30):
        ans = solve(N, edges, k)
        if ans == -1:
            print(-1)
            return
        
        for i in range(N):
            if ans[i] != -1:
                answers[i] += (1 << k) * ans[i]
    
    for i in range(N):
        print(answers[i])



if __name__ == "__main__":
    main()
0