并查集的原理和故事:如何理解并查集。
剑指 Offer II 116. 省份数量 - 力扣(LeetCode)
class Solution {
public:
int par[300];
//并查集查询+路径压缩
int find(int x){
if(par[x]!=x){
par[x] = find(par[x]);
}
return par[x];
}
int findCircleNum(vector<vector<int>>& isConnected) {
//并查集的初始化
for(int i=0;i<300;i++){
par[i]=i;
}
// 集合合并,题目中isConnected[i][i]=1,这个是必然成立,所以考虑将j点合并到i点。
for(int i =0;i<isConnected.size();i++){
for(int j= 0;j<isConnected[i].size();j++){
if(isConnected[i][j]==1){
par[find(j)] = find(i);
}
}
}
//最后判断有几个集合,父节点等于其自身就可以认为是一个集合
int res=0;
for(int i =0;i<isConnected.size();i++){
if(par[i]==i){
res++;
}
}
return res;
}
};
并查集在应用中还可以加入变形,即:加入对并查集内节点的判断。
int size[N]
int size[N];
//要求合并集合时,判断a,b是否已经在集合内,没有在集合内时
//将a集合内的元素个数,加入到b集合内部
if(find(a)!=find(b)){
size[find(a)]+=size[find(b)];
}