結果

問題 No.1420 国勢調査 (Easy)
ユーザー LyricalMaestro
提出日時 2025-04-19 13:33:27
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,388 bytes
コンパイル時間 483 ms
コンパイル使用メモリ 82,892 KB
実行使用メモリ 299,832 KB
最終ジャッジ日時 2025-04-19 13:33:51
合計ジャッジ時間 21,918 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20 TLE * 3 -- * 7
権限があれば一括ダウンロードができます

ソースコード

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]:
                    if values[w] == -1:
                        values[w] = y ^ values[v]
                        queue.append(w)
                    elif values[w] != y ^ values[v]:
                        return -1

    for i in range(N):
        if values[i] == -1:
            values[i] = 0
    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):
            answers[i] += (1 << k) * ans[i]
    
    for i in range(N):
        print(answers[i])



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