L3-003. 社交集群
在社交网络平台注册时,用户通常会输入自己的兴趣爱好,以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序,找出所有的集群。
输入格式:
输入的第一行给出正整数N(<=1000),即社交网络中的用户总数(则用户从1到N编号)。随后N行,每行按下列格式列出每个人的兴趣爱好:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是第i个人的兴趣的数量,hi[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
思路:不要管人的编号,要按照每个人的爱好用并查集建里联系,如果两个人之间有共同的爱好,那么他们的树一定是合并成一棵的,然后根据有几个根节点来统计集群数目,代码里的c数组是一个亮点,一定要理解它的含义和它存在的意义,它存在的是为了下面统计每个集群里的人数做准备的下面附上代码
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int inf = 1e3 + 10;
int id[inf];
int sum[inf];
int c[inf];
int find(int x)
{
//路径压缩,让当前节点指向它的父节点的父节点,也就是爷爷节点,可以推出的是,如果它的父节点就是根节点,该操作也是不会错误
while(id[x] != x){id[x] = id[id[x]];
x = id[x];
}
return x;
}
int main()
{
int num;
scanf("%d", &num);
for(int i = 0; i < inf; i ++)
id[i] = i;
for(int j = 0; j < num; j ++){
int n;
scanf("%d: ", &n);
scanf("%d", &c[j]);
for(int i = 1; i < n; i ++){
int t;
scanf("%d", &t);
if(find(c[j]) != find(t)){
id[find(c[j])] = t;
}
}
}
for(int i = 0; i < num; i ++){
sum[find(c[i])] ++;
}
int ans = 0;
for(int i = 0; i < inf; i ++){
if(sum[i]) ans ++;
}
printf("%d\n", ans);
sort(sum, sum + inf);
int N = 0;
for(int i = inf - 1; i >= 0; i --){
if(sum[i]){
if(N) printf(" ");
printf("%d", sum[i]);
N = 1;
}
}
printf("\n");
return 0;
}