PAT L3-003社交集群 (并查集）

最新推荐文章于 2021-04-23 21:04:43 发布

「已注销」

最新推荐文章于 2021-04-23 21:04:43 发布

阅读量294

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/PiscesCrab/article/details/69808050

本文介绍了一种寻找社交网络中兴趣相同用户集群的方法。通过利用并查集数据结构将具有相同兴趣爱好的用户归类到同一集群中，并统计各集群规模。

题目：

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

输出样例：

3
4 3 1

方法：将拥有同样爱好的人放入一个容器（vector）中，然后用并查集，将容器中所有人放入一个并查集中。

代码：

//By Sean Chen
#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std; 
int root[1005];
int findroot(int x)
{
	if (x==root[x])
		return x;
	return root[x]=findroot(root[x]);
}

void Uni (int a,int b)
{
	int aa=findroot(a),bb=findroot(b);
	if (aa!=bb)
		root[bb]=aa;
	return;
}

int cnt[1005];
int cmp(int a,int b)
{
	return a>b;
}
int main()
{
	vector <int> V[1005];
	
	int n,m,a;
	scanf("%d",&n);
	int M=-1;
	for (int i=1;i<=n;i++)
	{
		root[i]=i;
		scanf("%d",&m);
		getchar();
		for (int j=0;j<m;j++)
		{
			scanf("%d",&a);
			V[a].push_back(i);
			if (a>M)
				M=a;
		}
	}
	for (int i=1;i<=M;i++)
	{
		for (int j=1;j<V[i].size();j++)
		{
			Uni(V[i][j],V[i][0]);
		}
	}
	for (int i=1;i<=n;i++)
	{
		int temp=findroot(i);
		cnt[temp]++;
	}
	sort(cnt,cnt+1005,cmp);
	int p;
	for (int i=0;i<1005;i++)
	{
		if (cnt[i]==0)
		{
			p=i;
			break;
		}
	}
	printf("%d\n",p);
	for (int i=0;i<p;i++)
	{
		if (i!=p-1)
			printf("%d ",cnt[i]);
		else
			printf("%d",cnt[i]);
	}
	return 0;
}