L3-003 社交集群 (30分) C++并查集

题目描述

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

输入格式

输入在第一行给出一个正整数 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

---

分析

本题采用并查集的思想，把每一组输入的兴趣爱好的第一个设为父结点，后面的兴趣爱好向父结点靠拢，最后统计相同父结点的点个数然后输出即可。

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int n;
int intri[N];
int fa[N];
int find(int x)
{
	if(fa[x]!=x) fa[x]=find(fa[x]);
	return fa[x];
}
int main()
{
	for(int i=1;i<N;i++) fa[i]=i;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int k; cin>>k;
		char p; cin>>p; //抵消掉‘:’符号
		int f=0,num;
		while(k--)
		{
			cin>>num;
			if(f==0)
			{
				f=num;
				fa[f]=find(num);
				intri[i]=num;
			}
			else{
				int faa=find(num),fbb=find(f);
				fa[faa]=fbb;
			}
		}
	}
	for(int i=1;i<=n;i++) intri[i]=find(intri[i]);
	map<int,int> mp;
	for(int i=1;i<=n;i++)
	{
		mp[intri[i]]++;
	}
	cout<<mp.size()<<endl;
	vector<int> ans;
	for(auto x:mp) ans.push_back(x.second);
	sort(ans.begin(),ans.end());
	reverse(ans.begin(),ans.end());
	int flag=0;
	for(auto x:ans)
	{
		if(flag==1) cout<<" ";
		cout<<x;
		flag=1;	
	} 
	return 0;
}