PTA L3-003 社交集群

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

输入格式：

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

Ki​: hi​[1] hi​[2] ... hi​[Ki​]

其中Ki​(>0)是兴趣爱好的个数，hi​[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;
const int N=1000+10;
int n,m,s[N],res,f[N],flag,ans[N],now;
set<int>v[N];
int find(int x){			//查找 
	if(x!=s[x]) s[x]=find(s[x]);
	return s[x];
}
void merge(int x,int y){	//合并 
	x=find(x),y=find(y);
	if(x!=y) s[y]=x,f[x]+=f[y];
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) s[i]=i,f[i]=1;	//初始化 
	for(int i=1;i<=n;i++){
		cin>>m;cin.get();	//注意分号 
		for(int j=0;j<m;j++){	 
			int a;cin>>a;	
			v[i].insert(a);
			for(int k=1;k<i;k++){
				if(v[k].count(a)){	//有相同集群 
					merge(k,i);
					break;
				}	
			}
		}
	}
	for(int i=1;i<=n;i++) if(s[i]==i) res++,ans[now++]=f[i];
	cout<<res<<endl,sort(ans,ans+now);
	for(int i=now-1;i>=0;i--){
		cout<<ans[i];
		if(i!=0) cout<<" ";
	}	
}