L3-003. 社交集群

L3-003. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入的第一行给出正整数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个空格分隔，行首尾不得有多余空格。

输入样例：

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;
int pre[1001];

// 标准的并查集的2个函数
int find(int x) {
	int r = x;
	while (pre[r] != r)
		r = pre[r];
	// 路径合并
	int i = x, j;
	while (i != r) {

		j = pre[i];// 在改变上级之前用临时变量  j 记录下他的值 
		pre[i] = r;// 把上级改为根节点
		i = j;
	}
	return r;
}

void join(int x, int y) {
	int fx = find(x), fy = find(y);
	if (fx != fy)
		pre[fx] = fy;
}

int cmp(int a, int b) {
	return a > b;
}

int main()
{
	//freopen("data.txt", "r", stdin);
	int N, K;
	int course[1001] = { 0 };
	int count[1001] = { 0 };
	cin >> N;
	for (int i = 1; i <= N; i++)
		pre[i] = i;

	for (int i = 1; i <= N; i++) {
		scanf("%d: ", &K);
		for (int j = 0; j < K; j++) {
			int num;
			cin >> num;
			if (course[num] == 0)
				course[num] = i;
			// 将用户i加入并查集
			join(i, find(course[num]));
		}
	}

	// 遍历整个并查集，并查集值和下标相等时cnt++
	int cnt = 0;
	for (int i = 1; i <= N; i++) {
		if (pre[i] == i)
			cnt++;
	}
	cout << cnt << endl;

	// 统计集群用户数量
	for (int i = 1; i <= N; i++) {
		count[find(pre[i])]++;
	}

	// 排序
	sort(count, count + 1001,cmp);
	for (int i = 0; i < cnt; i++) {
		if (i == 0)
			cout << count[i];
		else
			cout << " " << count[i];
	}
	return 0;
}