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

#include <bits/stdc++.h>
using namespace std;
#define ll long long
int x[1010],fa[1010],ans[1010],sum=0,n,k,t;
string s;
int fin(int x)
{
	if(fa[x]==x) return x;
	return fa[x]=fin(fa[x]);
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;i++)
     fa[i]=i;
    for(int i=1;i<=n;i++)
    {
    	cin>>s;
    	k=0;
    	for(int j=0;j<s.length()-1;j++)
    	 k=k*10+s[j]-'0';
    	while(k--)
    	{
    		cin>>t;
    		if(x[t]==0) x[t]=i;
    		else 
			{
			 fa[fin(x[t])]=i;	
			 //cout<<x[t]<<" "<<fa[x[t]]<<endl;
			 x[t]=i;
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
	 ans[fin(i)]++;
	 if(ans[fin(i)]==1) sum++;	
    } 
	sort(ans+1,ans+n+1);
	cout<<sum<<endl;
	cout<<ans[n];
	for(int i=n-1,j=sum-1;i&&j;i--,j--)
	 cout<<" "<<ans[i];
    return 0;
}
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 终极编程指南 设计师:CSDN官方博客 返回首页