【PAT甲级】1114 Family Property (25分)（并查集）

题意：

输入一个正整数N（<=10000），接着输入N行每行包括一个人的ID和他双亲的ID以及他的孩子数量和孩子们的ID（四位整数包含前导零），还有他所拥有的房产数量和房产面积。输出一共有多少个块（所有有父子关系的点都视为处于同一块），以及每块的元素个数以及平均每人分得的房产数量和房产面积（保留三位小数），输出顺序按照房产面积降序，第二优先以ID升序。

AAAAAccepted code：

 #define HAVE_STRUCT_TIMESPEC
 #include<bits/stdc++.h>
 using namespace std;
 int ba[];
 int per[];
 int sets[],area[];
 int num[];
 int child[];
 int find_(int x){
     if(ba[x]==x)
         return x;

     return ba[x]=find_(ba[x]);
 };
 typedef struct{
     int id,num;
     double avgsets,avgarea;
 }st;
 st ans[];
 int vis[];
 int flag[];
 bool cmp(st a,st b){
     if(a.avgarea!=b.avgarea)
         return a.avgarea>b.avgarea;
     return a.id<b.id;
 }
 int main(){
     //ios::sync_with_stdio(false);
     //cin.tie(NULL);
     //cout.tie(NULL);
     for(int i=;i<;++i)
         ba[i]=i,num[i]=;
     int n;
     scanf("%d",&n);
     for(int i=;i<=n;++i){
         memset(flag,,sizeof(flag));
         int id;
         scanf("%d",&id);
         int x=,y=,z=;
         x=find_(id);
         int fa,mo;
         scanf("%d%d",&fa,&mo);
         if(fa!=-)
             y=find_(fa);
         if(mo!=-)
             z=find_(mo);
         int root=min(x,y);
         root=min(root,z);
         int k;
         scanf("%d",&k);
         int mn=;
         for(int j=;j<=k;++j){
             scanf("%d",&child[j]);
             mn=min(mn,find_(child[j]));
             vis[child[j]]=;
         }
         int tsets,tarea;
         scanf("%d%d",&tsets,&tarea);
         sets[x]+=tsets;
         area[x]+=tarea;
         if(mn<root)
             root=mn;
         if(x!=root)
             flag[x]=,num[root]+=num[x],sets[root]+=sets[x],area[root]+=area[x];
         if(y!=root&&y!=&&!flag[y])
             flag[y]=,num[root]+=num[y],sets[root]+=sets[y],area[root]+=area[y];
         if(z!=root&&z!=&&!flag[z])
             flag[z]=,num[root]+=num[z],sets[root]+=sets[z],area[root]+=area[z];
         ba[x]=root;
         if(y<)
             ba[y]=root;
         if(z<)
             ba[z]=root;
         for(int j=;j<=k;++j){
             int temp=find_(child[j]);
             ba[temp]=root;
             if(temp!=root&&!flag[temp]){
                 flag[temp]=;
                 num[root]+=num[temp];
                 sets[root]+=sets[temp];
                 area[root]+=area[temp];
             }
         }
         vis[id]=;
         if(fa!=-)
             vis[fa]=;
         if(mo!=-)
             vis[mo]=;
     }
     int cnt=;
     for(int i=;i<;++i){
         if(ba[i]==i&&vis[i]){
             ans[++cnt].id=i;
             ans[cnt].num=num[i];
             ans[cnt].avgsets=1.0*sets[i]/num[i];
             ans[cnt].avgarea=1.0*area[i]/num[i];
         }
     }
     sort(ans+,ans++cnt,cmp);
     printf("%d\n",cnt);
     for(int i=;i<=cnt;++i){
         printf("%04d %d %.3lf %.3lf",ans[i].id,ans[i].num,ans[i].avgsets,ans[i].avgarea);
         if(i<cnt)
             printf("\n");
     }
     return ;
 }