并查集之冗余连接

问题描述

在本问题中, 树指的是一个连通且无环的无向图。

输入一个图，该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组。每一个边的元素是一对[u, v] ，满足 u < v，表示连接顶点u 和v的无向图的边。

返回一条可以删去的边，使得结果图是一个有着N个节点的树。如果有多个答案，则返回二维数组中最后出现的边。答案边 [u, v] 应满足相同的格式 u < v。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/redundant-connection
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例1

输入: [[1,2], [1,3], [2,3]]
输出: [2,3]
解释: 给定的无向图为:

示例2

输入: [[1,2], [2,3], [3,4], [1,4], [1,5]]
输出: [1,4]
解释: 给定的无向图为:

解题思路

N个节点，M个关系，求解不相交集合查找和合并问题，直接套用并查集板子，并查集四要素

1. 初始化
2. 查找
3. 路径压缩
4. 合并

1 初始化

初始化每个节点的父节点（掌门），以示例2为例
pre[1] = 1; pre[2] = 2; pre[3] = 3; pre[4] = 4; pre[5] = 5;

2 查找 + 3 路径压缩

遍历数组，查找节点i的掌门是否跟节点j的掌门为同一个人（此处的掌门是另一个大神写的生动形象故事，文章最后会把大神博客地址贴出来），如果是同一个，则表明该关系是冗余的，因为数组前面已经有关系把此处的两个节点联通了，题目及测试用例已经假设不会出现不合理的情况，所以此处可以直接返回该关系。

for(int index=0; index<N; index++){
	//edges[index][0]是节点i,edges[index][1]是节点j
	if(find(edges[index][0]) == find(edges[index][1])){
		//返回该关系
	}
}

public int find(int x){
    if(pre[x]!=x) pre[x] = find(pre[x]);//递归进行路径压缩，也是并查集的模板代码
    return pre[x];
}

4 合并

并查集的模板代码，昨天才刚接触并查集，此处再手敲一遍，增加印象，哈哈哈~

public void union(int i, int j){
	//将其中的一个掌门置为另一个的掌门
	pre[find(i)] = find(j);
}

完整代码

class Solution {
    int[] pre;
    public int[] findRedundantConnection(int[][] edges) {
        //N个节点，M 个关系
        int N = edges.length; //N表示有多少个节点
        pre = new int[1001];

        //初始化
        for(int i=1; i<=N; i++)
            pre[i] = i;

        for(int i=0; i<N; i++){
            if(find(edges[i][0]) == find(edges[i][1])){
                return new int[]{edges[i][0], edges[i][1]};
            }
            union(edges[i][0], edges[i][1]);
        }
        return new int[2];
    }
    public void union(int x, int y){
        pre[find(x)] = find(y);
    }
    public int find(int x){
        if(pre[x]!=x) pre[x] = find(pre[x]);
        return pre[x];
    }
}

执行用时

大神的博客地址

超有爱的并查集~

结语

个人感觉并查集还是挺好用的，学习难度也不会很大，而且时间复杂度也比其他解法要小一些，之后也会坚持练习并查集相关的题目，争取早日把并查集学透。
本人所有博客仅用于学习记录，不做任何商业用途，如涉及侵权，还请联系删除，感谢阅读，欢迎留言，一起进步~