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

题目链接：https://www.patest.cn/contests/gplt/L3-003

本以为自己可以说已经会并查集了，但是这个题还是没有想到怎么做，还是不到家啊。

就不讲解了，会并查集的看代码都懂。

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int pre[10000];
int Find(int x){
    return x==pre[x]?x:Find(pre[x]);
}
void mix(int x,int y){
    int xx=Find(x);
    int yy=Find(y);
    if(xx!=yy){
        pre[xx]=yy;
    }
}
int t[10000];
int xin[10000];
int f[10000];
bool cmp(int a,int b){
    return a>b;
}
int main(){
    int n;
    for(int i=0;i<=1000;i++){
        pre[i]=i;
    }
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        int k;
        int t=-1;
        scanf("%d:",&k);
        for(int j=0;j<k;j++){
            int x;
            scanf("%d",&x);
            if(t==-1){
                f[i]=x;
                t=x;
            }
            else{
                mix(t,x);
                t=x;
            }
        }
    }
    for(int i=0;i<=1000;i++){
        t[Find(f[i])]++;
    }
    int top=0;
    for(int i=0;i<=1000;i++){
        if(t[i]!=0){
            xin[top++]=t[i];
        }
    }
    sort(xin+1,xin+top,cmp);
    cout<<top-1<<endl;
    for(int i=1;i<top;i++){
        printf(i==1?"%d":" %d",xin[i]);
    }
    cout<<endl;
    return 0;
}