Submission #31278649

---

Source Code Expand

class UnionFind:
    def __init__(self, n):
        self.par = [i for i in range(n)]
        self.siz = [1] * n
    def root(self, x):
        if self.par[x] == x:
            return x
        self.par[x] = self.root(self.par[x]) # 経路圧縮
        return self.par[x]
    def unite(self, x, y):
        x = self.root(x)
        y = self.root(y)
        if x == y:
            return
        if self.siz[x] > self.siz[y]: # マージテク
            self.par[y] = x
            self.siz[x] += self.siz[y]
        else:
            self.par[x] = y
            self.siz[y] += self.siz[x]
    def is_same(self, x, y):
        return self.root(x) == self.root(y)
    def size(self, x):
        return self.siz[self.root(x)]

n, q = map(int, input().split())
uf = UnionFind(n)
for i in range(q):
    t, u, v = map(int, input().split())
    if t == 0:
        uf.unite(u, v)
    if t == 1:
        print(int(uf.is_same(u, v)))

Submission Info

Judge Result