力扣684.冗余连接python解法

这道题我们猛然一看有点不懂,如何去判断最后两个是否连接呐?在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 []

并查集说明如下:
https://blog.csdn.net/qq_51718832/article/details/112340746

你可能感兴趣的:(力扣,python,leetcode)