PTA社交集群

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

输入格式：

输入在第一行给出一个正整数 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 <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> a[1000];
long long h[1000], s[1000];
int find(int x)
{
	long long m = x;
	while (m != h[m]) 
		m = h[m];
	h[x] = m;
	return m;
}
int main()
{
	int n;
	cin >> n;
	for (int i = 1; i <= n; ++i)
	{
		long long m, t;
		scanf("%lld:", &m);
		for (int j = 0; j < m; ++j)
		{
			cin >> t;;
			a[t].push_back(i);
		}
	}
	for (int i = 0; i < 1000; ++i)
	{
		h[i] = i;
		s[i] = 1;
	}
	for (int i = 1; i < 1000; ++i)
	{
		if (a[i].size() == 0)
			continue;
		int finda = find(a[i][0]);
		for (int j = 1; j < a[i].size(); ++j)
		{
			int findb = find(a[i][j]);
			if (finda != findb)
			{
				h[findb] = finda;
				s[finda] += s[findb];
			}
		}
	}

	long long cnt = 0;
	vector<int> ans;
	for (int i = 1; i <= n; ++i)
	{
		if (h[i] == i)
		{
			cnt++;
			ans.push_back(s[i]);
		}
	}
	sort(ans.begin(), ans.end());
	cout << cnt << endl;
	for (int i = cnt - 1; i >= 0; --i)
	{
		cout << ans[i];
		if (i != 0)
			cout << " ";;
	}
	return 0;
}