当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 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
思路:虽然并查集并起来的是兴趣的集合吗,但是c[]数组保存了每个编号所在的集合,用桶把兴趣集合相同的都储存下来,然后计算个数。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
#define MAXN 1010
int c[MAXN], f[MAXN];
int find(int x)
{
return x == f[x] ? x : f[x] = find(f[x]);
}
void merge(int x, int y)
{
x = find(x);
y = find(y);
if(x != y) f[x] = y;
}
int main()
{
int n, v, t, tt;
scanf("%d", &n);
for(int i = 0; i < 1010; i++) f[i] = i;
for(int i = 0; i < n; i++)
{
scanf("%d: ", &t);
scanf("%d", &tt);
c[i] = tt;
for(int j = 1; j < t; j++)
{
scanf("%d", &v);
merge(tt, v);
}
}
int num[1010] = {0}, cnt = 0;
for(int i = 0; i < n; i++)
num[find(c[i])] ++;
for(int i = 0; i < 1010; i++)
if(num[i])
cnt++;
sort(num, num + 1010, greater<int>());
printf("%d\n", cnt);
printf("%d", num[0]);
for(int i = 1; i < cnt; i++)
printf(" %d", num[i]);
printf("\n");
return 0;
}