UnionFind(相当)を10行で実装する - Qiita

import random import timeit import sys import platform class EasyUnionFind: """ dict と frozenset を使った実装。 """ def __init__(self, n): self._groups = {x: frozenset([x]) for x in range(n)} def union(self, x, y): group = self._groups[x] | self._groups[y] self._groups.update((c, group) for c in group) def groups(self): return frozenset(self._groups.values()) class UnionFind(object): """ 典型的なUnion-Find木に