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+];
 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=;i<n;++i){
         int t,h,Min,p;
         scanf("%d:",&t);
         for(int j=;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=;i<=N;++i)Find(i);
     for(int i=;i<n;++i)
         r[pre[a[i]]]++;
     sort(r,r+N,compare);
     int k;
     for(k=;k<N;k++)
         if(!r[k])break;
     printf("%d\n",k);
     printf("%d",r[]);
     for(int i=;i<k;++i)
         printf(" %d",r[i]);
     printf("\n");
     return ;
 }
 void Make(){
     for(int i=;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;
 }