Python并查集（Disjoint Set）

此昵称已存在，请勿使用

已于 2024-09-03 21:31:17 修改

阅读量385

点赞数 9

文章标签： python 数据结构

于 2024-02-29 12:58:02 首次发布

本文链接：https://blog.csdn.net/qq_28803395/article/details/136368031

版权

并查集可以使用列表、字典等来实现，用树的结构来表示，其中每个节点表示一个元素，树的根节点表示集合的代表元素（种类）

rank用来合并两个set，因为查找父节点时复杂度和深度正相关，所以思想是浅的set合并到深的set上去，尽量使树的结构扁平。在这段代码中，rank的意义是该点所处的深度。

class DisjointSet:
    def __init__(self, n):
        self.root = [i for i in range(n)]
        self.rank = [1] * n

    def find(self, x):
        if self.root[x] != x:
            self.root[x] = self.find(self.root[x])
        return self.root[x]

    def union(self, x, y):
        x_root = self.find(x)
        y_root = self.find(y)
        if x_root != y_root:
            if self.rank(x_root) <= self.rank(y_root):
                self.root[x_root] = y_root
                if self.rank(x_root) == self.rank(y_root):
                    self.rank[y_root] += 1
            else:
                self.root[y_root] = x_root