[并查集] 社交集群 天梯L3-003

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

输出样例：

3
4 3 1

#include <bits/stdc++.h>
using namespace std;
int p[1001];
int q[1001];
int find(int x)  //查找根节点
{
	if(p[x]==x) return x;
	return p[x]=find(p[x]);
}
void hebing(int a,int b)  //合并两树
{
	int fa=find(a);int fb=find(b);
	if(fa!=fb) p[fa]=fb;
}
int main()
{
	int n;while(cin>>n)
	{
		for(int i=1;i<=1000;i++)
			p[i]=i;
		for(int i=1;i<=n;i++)
		{
			int k;cin>>k;
			char c;cin>>c;
			int a;cin>>a;
			p[a]=find(a);  //根节点
			for(int j=2;j<=k;j++)
			{
				int b;cin>>b;
				hebing(a,b);
				p[b]=find(b);
			}
			q[i]=find(a);
			//第i个人 以兴趣的根来标记不同人
		}
		int sum[1001]={0};
		for(int i=1;i<=n;i++)
			sum[find(q[i])]++;
		//根为find(q[i])的人数 原未写find出错
		sort(sum,sum+1001,greater<int>());
		int t; //计集群数 排序后为0则停止
		for(t=0;t<=n;t++) if(!sum[t]) break;
		cout<<t<<endl<<sum[0];
		for(int i=1;i<t;i++)
			cout<<' '<<sum[i];
		cout<<endl;
	}
	return 0;
}