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 <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
#define LL long long
#define lt x<<1
#define rt x<<1|1
const int maxn = 1e5+111;
const int inf = 0x3f3f3f3f;
int val[1111];
int fa[1111];
char s;
int find(int x)
{
  return fa[x]==x?x:fa[x]=find(fa[x]);
}
void join(int x,int y)
{
  int xx=find(x);
  int yy=find(y);
  if(yy!=xx) fa[yy]=xx;
}
int num[1111];
int main()
{
  int n;
  scanf("%d",&n);
  for(int i=1;i<=n;i++) fa[i]=i;
  for(int i=1;i<=n;i++){
    int k;scanf("%d",&k);
    scanf("%c",&s);
    for(int j=1;j<=k;j++){
      int x;scanf("%d",&x);
      if(val[x]==0) val[x]=i;
      else{
        join(val[x],i);
      }
    }
  }
  for(int i=1;i<=n;i++){
    num[find(i)]++;
  }
  int ans[1111];
  int cnt=0;
  for(int i=1;i<=n;i++){
    if(num[i]>=1){
      ans[++cnt]=num[i];
    }
  }
  printf("%d\n",cnt);
  sort(ans+1,ans+1+cnt);
  for(int i=cnt;i>=1;i--){
    printf("%d%c",ans[i],i==1?'\n':' ');
  }

  return 0;
}