[路飞]_前端算法第八十七弹-冗余连接

222 阅读2分钟

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

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

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

示例 1:

**输入:** edges = [[1,2], [1,3], [2,3]]
**输出:** [2,3]

示例 2:

**输入:** edges = [[1,2], [2,3], [3,4], [1,4], [1,5]]
**输出:** [1,4]

由题意可知,上述图中会有一条边是多余的,导致该图无法形成一棵树,我们的目的就是找到这多余的一条边,由于可以去掉很多种边的情况,我们取最后一个,也就是说,我们可以一步一步的把这可树画出来,因为最开始各个点都是不相连的,由实例二可知,最开始我们将1、2点相连,然后是2、3点,这便是一个典型的并查集问题,此时得到的1、2、3为一个集合中,我们只需要找到最后加入集合中时,此时两点都已经在集合中了,那就证明这条边是多余的,因为此时该两点已经相连。

我们依旧需要先创建一个并查集

class UnionSet {
    constructor(n) {
        this.fa = new Array(n + 1)
        this.n = n
        for (let i = 0; i < n; i++) {
            this.fa[i] = i
        }
    }
    get(x) {
        return this.fa[x] = (this.fa[x] == x ? x : this.get(this.fa[x]))
    }
    merge(a, b) {
        this.fa[this.get(a)] = this.get(b)
    }
}

我们需要判断数组中每个元素中的两个点是否已经都在集合中,如果在,返回当前数组,如果不在,将两点加入到集合中。

var findRedundantConnection = function (edges) {
    let len = edges.length
    let u = new UnionSet(len)
    for (let i = 0; i < len; i++) {
        let e = edges[i]
        if (u.get(e[0]) == u.get(e[1])) return e
        u.merge(e[0], e[1])
    }
};