算法_并查集

最新推荐文章于 2024-08-16 10:27:08 发布

斕野清辰

最新推荐文章于 2024-08-16 10:27:08 发布

阅读量117

点赞数

分类专栏：算法文章标签：算法 python

本文链接：https://blog.csdn.net/u012891582/article/details/127542200

版权

算法专栏收录该内容

10 篇文章 0 订阅

订阅专栏

并查集

【核心思想】在一个图中判断两个node是不是相连，通过判断两个node的最终根节点是否相同来判断
【预置条件】假如一个图中的node.value，从0开始，依次递增
【步骤】
1、判断node1和node2是否连通，就是判断两者的root是否相同
2、如果不相同就把两者相连，就是把两者的root node相连。

relation_root_list：值是当前nod的值，索引是根节点的值
‘’’

class DSU:
    def __init__(self, node_num):
        self.relation_root_list = [-1] * node_num

    def find_root(self, curr_node):
        while self.relation_root_list[curr_node] != -1:
            curr_node = self.relation_root_list[curr_node]
        return curr_node

    def connect_node(self, node1, node2):
        root1 = self.find_root(node1)
        root2 = self.find_root(node2)
        if root1 != root2:
            self.relation_root_list[root1] = root2
        else:
            print('已经是连接的')

input_list =  [[1,2], [2,3], [3,4], [1,4], [1,5]]
obj = DSU(len(input_list)+1)
for ele in input_list:
    obj.connect_node(ele[0], ele[1])
print(obj.relation_root_list)