并查集总结
并查集,先写好init,union和findfather
主要问题是 和谁并,怎么并
本题逻辑:
1、要合并的是人i
2、合并的根据是课程,如果你喜欢某课程h,就要合并到喜欢该课程的人中去,那么就需要合并到喜欢该课程的人的根结点。
3、因此,需要一个数组course[h]来保存喜欢h活动的人的编号(不论是谁),findfather(course[h])找到根结点。然后Union(i, findFather(course[h]))
findFather(course[t])就是喜欢这个活动的⼈所处的社交圈⼦的根结点,合并根结点和当前⼈的编号的结点
4、要求输出社交网络数和各网络人数,用一个isRoot数组,然后遍历isRoot[findfather(i)]++;以i为根结点的人数就统计出来了。
注:之前不懂的点在于,不明白那些兴趣没有重合是怎么因为社交网络凑在一起的,原因是:每次每个人的每个兴趣都进行了Union(i, findFather(course[h])),在union的操作里,把i的father(上一个兴趣的father)和course[h]的father(下一个兴趣的father)合在一起了。
//怎么看不懂
#include<stdio.h>
#include<algorithm>
using namespace std;
const int N=1010;
int father[N];
int isRoot[N]={0};
int course[N]={0};
int findFather(int x){
int a=x;
while(x!=father[x]) x=father[x];
while(a!=father[a]){
int z=a;
a=father[a];
father[z]=x;
}
return x;
}
void Union(int a,int b){
int faA=findFather(a);
int faB=findFather(b);
if(faA!=faB) father[faA]=faB;
}
void init(int n){
for(int i=1;i<=n;i++){
father[i]=i;
}
}
bool cmp(int a,int b){
return a>b;
}
int main(){
int n,k,h;
scanf("%d",&n);
init(n);
for(int i=1;i<=n;i++){
scanf("%d:",&k);
for(int j=0;j<k;j++){
scanf("%d",&h);
if(course[h]==0) course[h]=i;
Union(i,findFather(course[h]));
//当然,写Union(i,course[h]);效果是一样的
}
}
int ans=0;
for(int i=1;i<=n;i++){
isRoot[findFather(i)]++;
}
for(int i=1;i<=n;i++){
if(isRoot[i]!=0) ans++;
}
printf("%d\n",ans);
sort(isRoot+1,isRoot+n+1,cmp);
for(int i=1;i<=ans;i++){
printf("%d",isRoot[i]);
if(i<ans) printf(" ");
}
return 0;
}