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

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/994805053141925888

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

 题解：这道题既然是划分群体，必然是用并查集来解。不过稍微有点特殊的是，一个群体内并不一定都有一个共同的爱好，比如说样例中人数为3的那个群体，只要一个人与团体中另一个人有相同的爱好就行。那么我们用interset[i]记录喜欢爱好i的第一个人，后面的人同样喜欢这个爱好的与interset[i]合并即可，即Union(i,interest[x])。最后用cnt[find(i)]统计群体人数即可。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1010;
int par[maxn];
int interest[maxn];
int cnt[maxn];
bool cmp(int x,int y)
{
	return x>y;
}
void init()
{
	for(int i=1;i<maxn;i++)
	par[i]=i;
}
int find(int x)
{
	if(par[x]==x)
	return x;
	else
	return par[x]=find(par[x]); 
}
void Union(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y)
	return ;
	else
	par[x]=y;
}
bool same(int x,int y)
{
	return find(x)==find(y);
}
int main()
{
	init();
	int n,t,x;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d:",&t);
		for(int j=1;j<=t;j++)
		{
			scanf("%d",&x);
			if(interest[x]==0)
			interest[x]=i;
			Union(i,interest[x]);
		}
	}
	for(int i=1;i<=n;i++)
	{
		cnt[find(i)]++;
	}
	sort(cnt,cnt+maxn,cmp);
	int num=0;
	for(int i=0;i<maxn;i++)
	{
		if(cnt[i])
		num++;
	}
	printf("%d\n",num);
	printf("%d",cnt[0]);
	for(int i=1;i<num;i++)
	printf(" %d",cnt[i]);
	return 0;
}