PAT L2-007 家庭房产（并查集 + 结构体）

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

输入格式：
输入第一行给出一个正整数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
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
栈限制
8192 KB

思路：

• 处理输入时先根据序号最小优先的原则建好并查集并且保存输入。然后处理
• 面积和套数仅登记在户主名下
• 对于每个家庭：
  最小序号 = root
  总人数 = root为序号且出现过的人
  总面积 = root为序号且出现过的人名下的面积总和
  总套数 = root为序号且出现过的人名下的套数总和

代码：

#include <bits/stdc++.h>

using namespace std;

const int N = 10010;

int p[N];
typedef struct member{
    int num;
    int sets, area;
    int isExist;
} Member;
Member allInput[N];

typedef struct node{
    int no, cnt;
    int ss, sa;
    double as, aa;
} Node;
bool operator < (const Node &a, const Node &b){
    if(fabs(a.aa -b.aa) > 1e-5) return (a.aa -b.aa) > 1e-5;
    return a.no < b.no;
}
Node ans[N];

int find(int x){
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

void unite(int a, int b){
    int pa = find(a);
    int pb = find(b);
    if(pa > pb) p[pa] = pb;
    else p[pb] = pa;
}

int main(){
    int n;
    cin >> n;
    for(int i = 0; i <= 9999; i++){
        p[i] = i;
    }
    set<int> roots;
    vector<int> nos;
    for(int i = 0; i < n; i++){
        int no, fa, mo, k;
        cin >> no >> fa >> mo >> k;
        nos.push_back(no);
        allInput[no].isExist = 1;
        if(fa != -1){
            unite(no, fa);
            allInput[fa].isExist = 1;
        }
        if(mo != -1){
            unite(no, mo);
            allInput[mo].isExist = 1;
        }
        for(int j = 0; j < k; j++){
            int t;
            cin >> t;
            unite(t, no);
            allInput[t].isExist = 1;
        }
        cin >> allInput[no].sets >> allInput[no].area;
    }

    // for(int i = 0; i < nos.size(); i++){
    //     int no = nos[i];
    //     cout << no << ": " << find(no) << endl;
    // }
    
    for(int i = 0; i <= 9999; i++){
        if(allInput[i].isExist){
            int pa = find(i);
            roots.insert(pa);
            ans[pa].no = pa;
            ans[pa].cnt ++;
            ans[pa].ss += allInput[i].sets;
            ans[pa].sa += allInput[i].area;
        }
    }
    vector<Node> res;
    for(auto &it:roots){
        ans[it].as = (double)ans[it].ss / (double)ans[it].cnt;
        ans[it].aa = (double)ans[it].sa / (double)ans[it].cnt;
        res.push_back(ans[it]);
    }
    sort(res.begin(), res.end());
    cout << roots.size() << endl;
    for(int i = 0; i < roots.size(); i++){
        Node tmp = res[i];
        printf("%04d %d %.3lf %.3lf\n", tmp.no, tmp.cnt, (double)tmp.ss / (double)tmp.cnt, (double)tmp.sa / (double)tmp.cnt);
    }
    return 0;
}