L2-007 家庭房产 （25 分)（并查集）

【题目】

L2-007 家庭房产 （25 分)

给定每个人的家庭成员和其自己名下的房产，请你统计出每个家庭的人口数、人均房产面积及房产套数。

输入格式：

输入第一行给出一个正整数N（≤1000），随后N行，每行按下列格式给出一个人的房产：

编号 父 母 k 孩子1 ... 孩子k 房产套数 总面积

其中编号是每个人独有的一个4位数的编号；父和母分别是该编号对应的这个人的父母的编号（如果已经过世，则显示-1）；k（0≤k≤5）是该人的子女的个数；孩子i是其子女的编号。

输出格式：

首先在第一行输出家庭个数（所有有亲属关系的人都属于同一个家庭）。随后按下列格式输出每个家庭的信息：

家庭成员的最小编号 家庭人口数 人均房产套数 人均房产面积

其中人均值要求保留小数点后3位。家庭信息首先按人均面积降序输出，若有并列，则按成员编号的升序输出。

输入样例：

10
6666 5551 5552 1 7777 1 100
1234 5678 9012 1 0002 2 300
8888 -1 -1 0 1 1000
2468 0001 0004 1 2222 1 500
7777 6666 -1 0 2 300
3721 -1 -1 1 2333 2 150
9012 -1 -1 3 1236 1235 1234 1 100
1235 5678 9012 0 1 50
2222 1236 2468 2 6661 6662 1 300
2333 -1 3721 3 6661 6662 6663 1 100

输出样例：

3
8888 1 1.000 1000.000
0001 15 0.600 100.000
5551 4 0.750 100.000

【题解】

思路：用并查集处理亲戚关系，以编号最小的为根结点。用结构体a处理并存储家庭信息，用结构体b进行排序。

【代码】

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct p{
    int num,cou; //编号，人口数目
    int sum1,sum2; //房产总数，面积总和
    double ans1,ans2; //平均值
}a[maxn],b[maxn];
int pre[maxn]; //根结点
int vis[maxn]={0};
int house[maxn]={0};
int area[maxn]={0};
int Find(int x)
{
    return x==pre[x]?x:pre[x]=Find(pre[x]);
}
void join(int x,int y)
{
    x=Find(x),y=Find(y);
    if(x<y) pre[y]=x;
    else pre[x]=y;
}
bool cmp(p x,p y)
{
    if(x.ans2==y.ans2)
        return x.num<y.num;
    return x.ans2>y.ans2;
}
void init()
{
    int n; scanf("%d",&n);
    for(int i=0;i<10000;i++) pre[i]=i;
    for(int i=0;i<n;i++){
        int t,fa,ma;
        scanf("%d%d%d",&t,&fa,&ma);
        vis[t]=1;
        if(fa!=-1){
            vis[fa]=1;
            join(t,fa);
        }
        if(ma!=-1){
            vis[ma]=1;
            join(t,ma);
        }
        int k; scanf("%d",&k);
        while(k--){
            int x; scanf("%d",&x);
            join(t,x);
            vis[x]=1;
        }
        scanf("%d%d",&house[t],&area[t]);
    }
}
int main()
{
    init(); //处理出父子关系，存储房产套数和房产面积
    for(int i=0;i<10000;i++){
        if(vis[i]==1){ 
            int root=Find(i);
            a[root].cou++;
            a[root].sum1+=house[i];
            a[root].sum2+=area[i];
        }
    }
    int cnt=0;
    for(int i=0;i<10000;i++){
        if(vis[i]==1&&Find(i)==i){
            b[cnt].num=i;
            b[cnt].cou=a[i].cou;
            b[cnt].ans1=a[i].sum1*1.0/a[i].cou;
            b[cnt].ans2=a[i].sum2*1.0/a[i].cou;
            cnt++;
        }
    }
    printf("%d\n",cnt);
    sort(b,b+cnt,cmp);
    for(int i=0;i<cnt;i++)
        printf("%04d %d %.3lf %.3lf",b[i].num,b[i].cou,b[i].ans1,b[i].ans2);
    return 0;
}