class UnionFind: def __init__(self, n): self.par = list(range(n)) self.size = [1] * n def find(self, x): if self.par[x] != x: self.par[x] = self.find(self.par[x]) return self.par[x] def unite(self, x, y): x = self.find(x) y = self.find(y) if x == y: return False if self.size[x] < self.size[y]: x, y = y, x self.par[y] = x self.size[x] += self.size[y] return True def same(self, x, y): return self.find(x) == self.find(y) def get_size(self, x): return self.size[self.find(x)] #nはノードの数 #0-idx uf = UnionFind(N)