PythonでJOI難易度5を埋める #7
考えたこと
グラフを連結して最終的に連結してるグループの数を考える問題。グラフについて色々調べてみると、UnionFindというのがあるらしいので使う。
PythonでのUnionFindから、いらない部分を削いで書きました。
class UnionFind(): def __init__(self, n): self.n = n self.parents = [-1] * n def find(self, x): if self.parents[x] < 0: return x else: self.parents[x] = self.find(self.parents[x]) return self.parents[x] def union(self, x, y): x = self.find(x) y = self.find(y) if x == y: return if self.parents[x] > self.parents[y]: x, y = y, x self.parents[x] += self.parents[y] self.parents[y] = x def roots(self): return [i for i, x in enumerate(self.parents) if x < 0] def group_count(self): return len(self.roots()) n = int(input()) m = int(input()) fibers = [list(map(int,input().split())) for _ in range(m)] uf = UnionFind(n) for i in range(m): uf.union(fibers[i][0]-1,fibers[i][1]-1) print(uf.group_count()-1)