L3-003 社交集群 并查集

L3-003 社交集群 （30 分）

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

K​i​​: h​i​​[1] h​i​​[2] ... h​i​​[K​i​​]

其中K​i​​(>0)是兴趣爱好的个数，h​i​​[j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：

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<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
#define inf 0x3f3f3f3f
#define LL long long
int father[1005];
int find(int x){
	return father[x]==x?x:father[x]=find(father[x]);
}
void combine(int x,int y)
{
	father[find(x)]=find(y);
}
bool cmp(int x,int y)
{
	return x>y;
}
int main()
{
	int n,m,k,l,o,i,j,f;
	vector<int> v[1005];
	for(i=0;i<1005;i++)
	father[i]=i;	
	scanf("%d",&n);
	for(i=1;i<=n;i++)
	{
		scanf("%d:",&m);
		for(j=0;j<m;j++)
		{
			scanf("%d",&k);
			v[i].push_back(k);
		}
	}
	for(i=1;i<=n;i++)
	{
		for(j=i+1;j<=n;j++)
		{
			for(l=0;l<v[i].size();l++)
			{
				f=0;
				for(o=0;o<v[j].size();o++)
				{
					if(v[i][l]==v[j][o])
					{
						f=1;
						combine(j,i);
						break;
					}
				}
				if(f)break;
			}
		} 
	}
	int sum=0;
	map<int,int> ma;
	for(i=1;i<=n;i++)
	{
		if(father[i]==i)
		sum++;
		ma[find(father[i])]++;//一定要find哦！！！ 
		//printf("i:%d f[i]:%d\n",i,father[i]);
	}
	printf("%d\n",sum);
	map<int,int>:: iterator it;
	int s[1005],op=0;
	for(it=ma.begin();it!=ma.end();it++)
	{
		s[op++]=it->second;
	}
	sort(s,s+op,cmp);
	for(i=0;i<op-1;i++)
	printf("%d ",s[i]);
	printf("%d\n",s[op-1]);
}