L3-003. 社交集群(并查集)

L3-003. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入的第一行给出正整数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<iostream>
#include<algorithm>
#include<set>
#include<cstring>
using namespace std;
int f[1005],sum[1005],vis[1005];
int find(int x)
{
	if(f[x]!=x)
	f[x]=find(f[x]);
	return f[x];
}
bool cmp(int a,int b)
{return a>b;}
int main()
{
	memset(vis,0,sizeof(vis));
	memset(sum,0,sizeof(sum));
	for(int i=0;i<1005;i++)
	f[i]=i;
	int t;
	cin>>t;
	while(t--)
	{
		int n,a[1005],F=0;
		char s[5];
		scanf("%s",s);
		n=s[0]-'0';
		
		for(int i=0;i<n;i++)
		{
			cin>>a[i];
			if(vis[a[i]])//if此时人的兴趣之前已经出现，则将该兴趣圈的祖先作为此时的祖先 
			F=find(a[i]);
		}
		if(F==0)//如果当前人所有兴趣之前从未出现，则将第一个兴趣作为父节点 
			{F=a[0];sum[F]=1;} 
				else sum[F]++;//如果兴趣之前出现过，则此兴趣圈人数+1 
		for(int i=0;i<n;i++)
		{
			vis[a[i]]=1;
			int ff=find(a[i]);
			if(ff!=F)
			{
				f[ff]=F;
				if(sum[ff])//将此兴趣所在兴趣圈的人数转移至当前兴趣圈中， 
				{sum[F]+=sum[ff];sum[ff]=0;}
			}
		}
	}
		set<int> S;//用于去重 
		for(int i=0;i<1002;i++)
		if(vis[i])
		S.insert(find(f[i]));
		cout<<S.size()<<endl;
		
		sort(sum,sum+1002,cmp);
		for(int i=0;i<1002;i++)
			if(sum[i+1]) cout<<sum[i]<<" ";
				else {cout<<sum[i];break;}
}