LeetCode 684. Redundant Connection [模版]

题解

这题就是并查集。然后我想总结下并查集的写法。
主要实现两块，union & find。
设p[ i ]代表 i 的父类。
find() 主要就是查找i的父类集合

// 无修改
int find_1(int i){
	while(p[i]!=i) 
		i=p[i];
	return i;
}
// 压缩路径
int find_2(int i){
	if(p[i]!=i) 
		p[i]=find(p[i]);
	return p[i];
}

union() 要注意这个合并一定要把最头上的两点连起来

void union(int x, int y){
	if(find(x) != find(y){
		p[find(x)] = find(y);
		// p[ x ] = find(y) 不行 p[x] 不能保证一定修改成最终父点
	}
}

---

Code

class Solution {
public:
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        vector<int> p(edges.size()+1);
        for(int i=0;i<p.size();i++) p[i]=i;
        
        int a,b;
        vector<int> res;
        for(auto e:edges){
            a=e[0];
            b=e[1];
            if(find(p,a) == find(p,b)) res=e;// 第一个 已经连接的两点
            else p[find(p,a)] = find(p,b);
        }
        return res;
    }
    int find(vector<int> &p, int i){
        if(p[i]==i) return i;
        return p[i]=find(p,p[i]);
    }
};