Pythonで螺旋本！螺旋本でPython！（１４章〜） - Qiita

class UnionFindTree(): def __init__(self, n): #木全体の要素数 self.n = n #root[x]<0ならそのノードが根でありその値が木の要素数 self.root = [-1] * (n+1) #ランク self.rank = [1] * (n+1) def find_root(self,x): if self.root[x] < 0: return x else: self.root[x] = self.find_root(self.root[x]) return self.root[x] def unite(self,x,y): x = self.find_root(x) y = self.find_root(y) if x == y: return elif self.rank[x] > self.rank[y]: self.roo