Leetcode 684. 冗余连接（并查集模板）

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

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

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

---

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

---

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

---

【思路】：并查集。

如果缺少某一条边的情况下，仍能让所有节点都处于一个集合内，那么就说明这条边是冗余的。

---

【代码】：

/**
 * @param {number[][]} edges
 * @return {number[]}
 */
//用并查集。每次尝试删除一条边，然后看整个图是否仍然都是一个parents
//一直循环下去


//return parent's val
var getParent = function(node, parent){
    while(parent[node] != node){
        node = parent[node];
    }
    return node;
}

//判断两个节点是否是一个集
var isSameGroup = function(node1, node2, parent){
    let p1 = getParent(node1, parent);
    let p2 = getParent(node2, parent);
    if(p1 == p2)    return true;
    return false;
}

//合并
var union = function(node1, node2, parent){
    let p1 = getParent(node1, parent);
    let p2 = getParent(node2, parent);
    parent[p2] = p1;
}

var findRedundantConnection = function(edges) {
    var n = edges.length;
    var parent = new Array(n + 1).fill(-1);
    for(let i = 0;i < parent.length;i++){
        parent[i] = i;  //默认父亲是自己
    }

    var res = [];
    //每次锁定一个边j
    for(let j = 0;j < edges.length;j++){

        //每次重新建一个图
        for(let i = 0;i < edges.length;i++){
            if(i == j)  continue;   //每次少建一条边
            let edge = edges[i];
            if(!isSameGroup(edge[0], edge[1], parent)){
                union(edge[0], edge[1], parent);
            }
        }

        //是否所有点的祖先一样
        let finalParent = getParent(1, parent);
        let bool = true;
        for(let i = 2;i < parent.length;i++){
            if(finalParent != getParent(i, parent)){
                bool = false;
                break;
            }
        }
        if(bool){
            res = [edges[j][0], edges[j][1]];
        }

        //fresh
        for(let i = 0;i < parent.length;i++){
            parent[i] = i;  //默认父亲是自己
        }
    }
    return res;
};