L3-1 社交集群 (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

额外注意一个点：同一个数字，在不同的数组中代表不同的意思；

在不同的函数中也有可能代表不同的意思 

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 1010;

int x[N], pre[N], ans[N], sum , n, k, t;
int cnt;

int find(int x)
{
	if (pre[x] != x) pre[x] = find(pre[x]);
	return pre[x];
}
int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++) 
		pre[i] = i;

	for (int i = 1; i <= n; i++)
	{
		cin >>cnt;
		getchar();
		while (cnt--)
		{
			cin >> t;
			//x[]存的是之前有没有读进去这个数
			if (x[t] == 0) x[t] = i;
			else
			{
				//将这个数字之前的父节点，作为当前节点的儿子
				pre[find(x[t])] = i;
				x[t] = i;
			}
		}
	}

	//统计根节点的个数以及每个树中的人数
	for (int i = 1; i <= n; i++)
	{
		ans[find(i)]++;//i所在的父节点的人数
		if (ans[find(i)] == 1)sum++;//连通块的个数        /
		 

	}

	sort(ans + 1, ans + n + 1,greater<int>());

	cout << sum << endl;

	for (int i = 1; i <sum; i++)
		cout << ans[i] << ' ';
	cout << ans[sum];

	return 0;
}