[路飞]_684. 冗余连接

78 阅读1分钟

解题思路

树可以看成是一个连通且 无环 的 无向 图。

给定往一棵 n 个节点 (节点值 1~n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间,且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ,edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。

请找出一条可以删去的边,删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案,则返回数组 edges 中最后出现的边。

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/re… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

并查集的思路 遍历数组,判断两个节点是否已经在一个集合中,如果是就直接返回这个节点,否则就建立关系

代码

/**
 * @param {number[][]} edges
 * @return {number[]}
 */
 class UnionSet {
    constructor(n) {
        this.fa = new Array(n + 1)
        for (let i = 0; i <= n; i++) {
            this.fa[i] = i
        }
    }
    get(x) {
        let fa = this.fa
        return fa[x] = (fa[x] === x ? x : this.get(fa[x]))
    }
    merge(a, b) {
       this.fa[this.get(a)] = this.get(b)
    }
}
var findRedundantConnection = function(edges) {
    let n = edges.length
    let u = new UnionSet(n)
    for (let i = 0; i < n; i++) {
        let item = edges[i]
        if (u.get(item[0]) === u.get(item[1])) return [item[0], item[1]]
        u.merge(item[0], item[1])
    }
};