

def union(self, a, b):
    pi1 = self._pi(a)
    pi2 = self._pi(b)
    if (pi1 != pi2):
        if (self.sz[pi1] > self.sz[pi2]):
            (pi1, pi2) = (pi2, pi1)
        self.pi[pi1] = pi2
        self.sz[pi2] += self.sz[pi1]
        self.count -= 1

dataset/ETHPy150Open algorhythms/LintCode/Number of Islands II.py UnionFind.union/original