L3-003. 社交集群

L3-003. 社交集群

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

查并集

与L2-007(家庭房产)类似，都是采用了并查集的算法，相对来说这题处理起来更简单一点。这里我维护了最小的h[i]，便于查找。

代码如下：

#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 1005
using namespace std;
int a[N];//存放各组h[k]的最小值
int pre[N+1];
int r[N];//存放结果sum
void Make();
int Find(int n);
void Uion(int x,int y);
bool compare(int x,int y){
    return x>y;
}
int main(void){
    freopen("in.txt","r",stdin);
    int n;
    Make();
    scanf("%d",&n);
    for(int i=0;i<n;++i){
        int t,h,Min,p;
        scanf("%d:",&t);
        for(int j=0;j<t;++j){
            scanf("%d",&h);
            if(!j){
                Min=h;
                p=h;
            }else{
                Min=min(Min,h);
                Uion(h,p);
            }
        }
        a[i]=Min;
    }
    for(int i=1;i<=N;++i)Find(i);
    for(int i=0;i<n;++i)
        r[pre[a[i]]]++;
    sort(r,r+N,compare);
    int k;
    for(k=0;k<N;k++)
        if(!r[k])break;
    printf("%d
",k);
    printf("%d",r[0]);
    for(int i=1;i<k;++i)
        printf(" %d",r[i]);
    printf("
");
    return 0;
}
void Make(){
    for(int i=0;i<=N;++i)
        pre[i]=i;
}
int Find(int n){
    if(n!=pre[n])pre[n]=Find(pre[n]);
    return pre[n];
}
void Uion(int x,int y){
    int k1=Find(x),k2=Find(y);
    k1<k2?pre[k2]=k1:pre[k1]=k2;
}