L2-007. 家庭房产

L2-007. 家庭房产

题目链接：https://www.patest.cn/contests/gplt/L2-007

并查集

初学，看这题的时候完全没有什么好的想法，参考了@yinzm的blog用BFS做了一遍，错了三个测试点，找了一下午没找出来，另寻他法。看到了并查集这个概念，做了一两题稍微掌握了一下，尝试做这题。

代码如下：

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #define EPS 1e-8
 #define N 10000
 using namespace std;
 struct node{
     int house,area;
 };
 struct res{
     int num;
     int people;
     int house;
     int area;
     double rh,ra;
 };
 res r[];
 int k;
 node a[N];
 int pre[N];
 bool mark[N];
 void Make(){
     for(int i=;i<N;++i)pre[i]=i;
 }
 int Find(int x){
     if(x!=pre[x])
         pre[x]=Find(pre[x]);
     return pre[x];
 }
 void uion(int x,int y){
     int k1=Find(x),k2=Find(y);
     k1<k2?pre[k2]=k1:pre[k1]=k2;
 }
 bool compare(res x,res y){
     if(fabs(x.ra-y.ra)<EPS)return x.num<y.num;
     else return x.ra>y.ra;
 }
 int main(void){
     freopen("in.txt","r",stdin);
     int n;
     scanf("%d",&n);
     Make();
     for(int i=;i<n;++i){
         int me,father,mother,num,kid[],house,area;
         scanf("%d%d%d%d",&me,&father,&mother,&num);
         mark[me]=;
         if(father!=-){
             uion(me,father);
             mark[father]=;
         }
         if(mother!=-){
             uion(me,mother);
             mark[mother]=;
         }
         for(int i=;i<num;++i){
             scanf("%d",&kid[i]);
             mark[kid[i]]=;
             uion(me,kid[i]);
         }
         scanf("%d%d",&house,&area);
         a[me].house=house;
         a[me].area=area;
     }
     for(int i=;i<N;++i)
         Find(i);
     int p;
     for(int i=;i<N;++i){
         if(mark[i]){
             if(k==){
                 r[k].num=pre[i];
                 r[k].people++;
                 r[k].house=a[i].house;
                 r[k].area=a[i].area;
                 k++;
             }else{
                 bool flag=true;
                 for(p=;p<k;++p){
                     if(r[p].num==pre[i]){
                         flag=false;
                         break;
                     }
                 }
                 if(flag){
                     r[k].num=pre[i];
                     r[k].people++;
                     r[k].house=a[i].house;
                     r[k].area=a[i].area;
                     k++;
                 }else{
                     r[p].people++;
                     r[p].house+=a[i].house;
                     r[p].area+=a[i].area;
                 }
             }
         }
     }
     for(int i=;i<k;++i){
         r[i].rh=r[i].house*1.0/r[i].people;
         r[i].ra=r[i].area*1.0/r[i].people;
     }
     printf("%d\n",k);
     sort(r,r+k,compare);
     for(int i=;i<k;++i)
         printf("%04d %d %.3lf %.3lf\n",r[i].num,r[i].people,r[i].rh,r[i].ra);
     return ;
 }