并查集必刷题pat1107 Social Clusters

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int N=1010;

int isRoot[N];
int father[N]; 
int course[N]; 

//n个节点的初始化 
void init(int n){
	
	for(int i=1;i<=n;i++){
		father[i]=i;//默认父节点是本身 
		isRoot[i]=0;
	}
}

//寻找x的父节点 并且采用路径压缩的方式 
int findFather(int x){
	int a=x;//先保存起来x节点 后面x改变了 
	while(x!=father[x])
		x=father[x];
	//循环结束的时候 x是原本x的根节点 
	
	
	while(a!=father[a]){
		int z=a;
		a=father[a];//保存a的父节点
		father[z]=x;//设置z的根节点是x 
		
	}
	
	return x;//返回根节点 
} 


//合并a b所在的集合 
void Union(int a,int b){
	
	int fatherA=findFather(a);
	int fatherB=findFather(b);
	
	if(fatherA!=fatherB){

		father[fatherA]=fatherB; 
	}
		
}

bool cmp(int a,int b){
	return a>b;
}

int main(void){
	
	int n,k,h;
	
	scanf("%d",&n);
	init(n);
	for(int i=1;i<=n;i++){
		scanf("%d:",&k);
		for(int j=1;j<=k;j++){//喜欢k个活动 
			scanf("%d",&h);//输入对应的活动号 
			if(course[h]==0){//h活动还没被人喜欢
				course[h]=i;//设置i号喜欢h活动 
			}	
			
			Union(i,findFather(course[h]));
		}
	
	} 
	 
	for(int i=1;i<=n;i++){
	
		isRoot[findFather(i)]++;//i所在集合的根节点被作为根节点一次 加1说明有一个子节点 
		
	}
	
	 
	
	//sort(isRoot,isRoot+n+1); 
	//统计有几组
	int ans=0; 
	for(int i=1;i<=n;i++){
		if(isRoot[i]!=0){
			ans++;
		}	
	}
	printf("%d\n",ans); 
	//排序 从大到小 
	sort(isRoot+1,isRoot+n+1,cmp);//isRoot数组从1号下标开始存放 
	for(int i=1;i<=ans;i++)
	{
		printf("%d",isRoot[i]);
		if(i!=ans) printf(" ");//除了最后一个数字之外 其余在结果后输出空格 
	}
	return 0;
}