【算法——Python实现】并查集及优化

class UnionFind(object):
    """
    Quick Find
    """
    def __init__(self, n):
        # 传入列表的数量n，列表id初始化为每个元素代表一个组
        self.count = n
        self.id = range(n)

    def find(self, p):
        # 查找元素对应的组
        if p >= 0 and p < self.count:
            return self.id[p]

    def isConnected(self, p, q):
        # p q两个元素是否连接，返回两个元素的组是否相等
        return self.find(p) == self.find(q)

    def unionElements(self, p, q):
        # p q的并
        pID = self.find(p)
        qID = self.find(q)
        if pID == qID:
            # pq的组相等，不需要并，直接返回
            return None
        i = 0
        while i < self.count:
            if self.id[i] == pID:
                self.id[i] = qID
            i += 1

---

class UnionFind2(object):
    """
    Quick Union，每个元素的组指向（等于）父节点的元素，根节点指向（等于）自身
    """
    def __init__(self, n):
        # 传入列表的数量n，列表id初始化为每个元素代表一个组
        self.count = n
        self.parent = range(n)
        self.sz = [1