题解
这题就是并查集。然后我想总结下并查集的写法。
主要实现两块,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]);
}
};