团体程序设计天梯赛 L3-003 社交集群

L3-003 社交集群

分数 30

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

输入格式：

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

题解 

并查集，枚举每一种爱好，如果这种爱好有超过1个人，那么就把这些人合并，最后合并完再遍历一遍，就能找到不同的集群，用优先队列来进行存储，非增输出。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
int n;
//vector<int> g[1005];
map<int,vector<int> > mp;
map<int,int> ans;
int f[1005];
priority_queue<int,vector<int> ,less<int> > q;
int find(int x)//并查集模板
{
	if(x==f[x])
	{
		return x;
	}
	f[x]=find(f[x]);
	return f[x];
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		f[i]=i;
		int k;
		//cin>>k;
		scanf("%d:",&k);
		for(int j=1;j<=k;j++)
		{
			int x;
			cin>>x;
			//g[i].push_back_back(x);
			mp[x].push_back(i);//放到每个爱好里
		}
	}
	for(auto k:mp)//遍历爱好
	{
		if(k.second.size()>1)
		{
			vector<int> temp=k.second;
			int u=find(temp[0]);
			for(int i=1;i<temp.size();i++)
			{
				int v=find(temp[i]);
                //合并
				if(u!=v)
				{
					f[v]=u;
				}
			}
			//int u=find(k.second[0]);
			
		}
	}
	for(int i=1;i<=n;i++)
	{
		ans[find(i)]++;
	}
	for(auto k:ans)
	{
		q.push(k.second);
	}
	int len=q.size();
	int op=0;
	cout<<len<<endl;
	while(!q.empty())//优先队列非递增输出
	{
		op++;
		if(op<len)
		{
			cout<<q.top()<<" ";
		}
		else
		{
			cout<<q.top()<<endl;
		}
		q.pop();
	}
	return 0;
}