結果

問題 No.1420 国勢調査 (Easy)
ユーザー gorugo30
提出日時 2021-03-05 22:11:12
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,690 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 82,312 KB
実行使用メモリ 98,772 KB
最終ジャッジ日時 2024-10-07 02:15:11
合計ジャッジ時間 4,292 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFind:
    def __init__(self, N):
        self.par = [-1] * N
        self.size = [1] * N
    def root(self, x):
        while self.par[x] >= 0:
            x = self.par[x]
        return x
    def union(self, x, y):
        rx = self.root(x)
        ry = self.root(y)
        if rx == ry:
            return
        if -self.par[rx] == -self.par[ry]:
            self.par[ry] -= 1
            self.par[rx] = ry
            self.size[ry] += self.size[rx]
            return
        if -self.par[rx] > -self.par[rx]:
            rx, ry = ry, rx
        self.par[rx] = ry
        self.size[ry] += self.size[rx]
    def find(self, x, y):
        return self.root(x) == self.root(y)
    def getsize(self, x):
        return self.size[self.root(x)]

N, M = map(int, input().split())
ufts = [UnionFind(2 * N) for i in range(30)]
for hito in range(M):
    A, B = map(int, input().split())
    A -= 1
    B -= 1
    Y = int(input())
    for i in range(30):
        if Y & 1:
            if ufts[i].find(A, B):
                print(-1)
                exit()
            ufts[i].union(A, B + N)
            ufts[i].union(A + N, B)
        else:
            if ufts[i].find(A + N, B):
                print(-1)
                exit()
            ufts[i].union(A, B)
            ufts[i].union(B, A)
        Y >>= 1
pow2 = 1
X = [0] * N
for k in range(30):
    x = [0] * N
    for i in range(N):
        if ufts[k].getsize(i) == 1:
            continue
        r = ufts[k].root(i)
        if r < N:
            x[i] = x[r]
            X[i] += pow2 * (x[r])
        else:
            r -= N
            x[i] = 1 ^ x[r]
            X[i] += pow2 * (1 ^ x[r])
    pow2 *= 2
print(*X, sep = "\n")
0