力扣684.冗余连接python解法

Leosaf

于 2021-01-15 15:07:03 发布

阅读量186

点赞数

分类专栏：力扣

本文链接：https://blog.youkuaiyun.com/qq_51718832/article/details/112669049

版权

python leetcode

力扣专栏收录该内容

67 篇文章

订阅专栏

这篇博客探讨了一种针对树状图题目，利用并查集算法来判断数组元素是否联通的方法。在给定的代码实现中，通过建立并查集，并实现`find`和`merge`函数，有效地检查边是否将数组的节点连接在一起。当存在一个数组使得所有节点联通时，该算法能找出这样的数组。并查集是一种高效的数据结构，适用于处理集合的连接与查询操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题我们猛然一看有点不懂，如何去判断最后两个是否连接呐？在547题中，也是以树状图给的题目，我们利用并查集进行连接，这道题我们是否也可以用并查集？
题目中写出了必定有一个数组他是联通整个数组的，那么用并查集来说的话，就是两者的父辈就变得相同了！我们这里不必写完整的并查集，只用连接和查找部分即可解决

        x = {}
        #建立并查集
        n = len(edges)
        def find(p):
            x.setdefault(p, p)
            if p != x[p]:
                x[p] = find(x[p])
            return x[p]
        def merge(a, b):
            x[find(a)] = find(b)
        for i in range(n):
            if find(edges[i][0]) != find(edges[i][1]):
                merge(edges[i][0], edges[i][1])
            else:
                return edges[i]
        return []