UnconFind.py

class UnionFind:
    def __init__(self, N):
        self.rank = [0]*N
        self.par = list(range(N))
 
    def find(self, x):
        if x != self.par[x]:
            self.par[x] = self.find(self.par[x])
        return self.par[x]
    
    def unite(self, x, y):
        x, y = self.find(x), self.find(y)
        if(self.rank[x] > self.rank[y]):
            self.par[y] = x
        else:
            self.par[x] = y
        if(self.rank[x] == self.rank[y]):
            self.rank[y] += 1