解题思路
树可以看成是一个连通且 无环 的 无向 图。
给定往一棵 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])
}
};