L3-003 社交集群 (30分)（并查集和set的配合解题）

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

输入格式：
输入在第一行给出一个正整数 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<bits/stdc++.h>
using namespace std;
int find(int a,int father[])
{
	if(father[a]==a) return a;
	else return father[a]=find(father[a],father);
} 
set<int> se; 
void mix(int a,int b,int father[])
{
	a=find(a,father);
	b=find(b,father);
	se.insert(a);
	se.insert(b);
	if(a==b) return;
	else father[b]=a;
	se.erase(b);
}
int main()
{
	int n,m,x,y,t;
	int v[1001]={0};
	scanf("%d",&n);
	int g[n][1001]={0};
	set<int> r;
	int father[1001];
	for(int i=0;i<1001;i++) father[i]=i;
	for(int i=0;i<n;i++){
		scanf("%d: %d",&m,&x);
		t=0;
		g[i][t++]=x;
		mix(x,x,father);
		
		for(int j=1;j<m;j++){
			scanf("%d",&y);
			mix(x,y,father);
			g[i][t++]=y;
		}
	}
	cout<<se.size()<<endl;
	for(int i=0;i<n;i++){
		for(int j=0;g[i][j]!=0;j++){
			if(se.count(find(g[i][j],father))){
				v[find(g[i][j],father)]++;
				break;
			}
		}
	}
	sort(v,v+1001,greater<int>());
	int f=0;
	for(int i=0;v[i]!=0;i++){
		if(f==1) printf(" ");
		printf("%d",v[i]);
		f=1;
	}
}