PTA 团体程序设计天梯赛-练习集 - L2-007 家庭房产 （并查集，结构体）

惊天大恶心
对于测试点2，3，4错误的：记得有0000这个编号。

---

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

输入格式：
输入第一行给出一个正整数$N（\le 1000）$，随后 $N$ 行，每行按下列格式给出一个人的房产：
编号 父 母 k 孩子1 ... 孩子k 房产套数 总面积
其中编号是每个人独有的一个4位数的编号；父和母分别是该编号对应的这个人的父母的编号（如果已经过世，则显示-1）；$k（0\le k\le 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

---

思路比较简单，首先这个关系树就要用并查集来维护，其次对于题目要求的排序方式就只有定义结构的的解决方法（只有这样才能保持所有数值捆绑在一起）。

CODE:

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5;

struct Node
{
    int id, cnt;
    double House, S;
} node[N];  //定义结构体数组存所有输入状态

int p[N];       //并查集板子
int find(int x)
{
    if (p[x] != x)
        p[x] = find(p[x]);
    return p[x];
}
void merge(int a, int b) 
{
    a = find(a), b = find(b);
    if (a > b) p[a] = b;    //直接尽量把编号数值更小的当作上级
    else p[b] = a;
}

bool vis[N];    //标记一个编号是否出现过

int main()
{
    for (int i = 1; i < N; i++) //初始化并查集
        p[i] = i;

    int n;
    cin >> n;
    while (n--)
    {
        int id, fa, mo;
        cin >> id >> fa >> mo;
        vis[id] = 1;

        if (fa != -1)   //只有活着的才在关系里
        {
            merge(id, fa);
            vis[fa] = 1;
        }

        if (mo != -1)
        {
            merge(id, mo);
            vis[mo] = 1;
        }

        int k;
        cin >> k;
        while (k--)
        {
            int x;
            cin >> x;
            vis[x] = 1;
            merge(id, x);
        }

        double house, s;cin >> house >> s;
        //读入房产信息，并且直接赋给当前的根节点即可
        id = find(id);
        node[id] = {id, 0, node[id].House + house, node[id].S + s};
    }

    for (int i = 0; i < N; i++)
    {
        if (vis[i])
        {
            node[find(i)].cnt++;    //如果这个人出现过一次，那么关系树中的根节点就要加上一个人
                                    //这样处理比较方便，如果边合并边处理很麻烦

            if (find(i) != i)   //不能自己加自己的信息
            {
                node[find(i)].House += node[i].House;
                node[find(i)].S += node[i].S;
            }
        }
    }

    vector<Node> res;   //用vector存答案，省空间
    for (int i = 0; i < N; i++)
    {
        if (vis[i] && p[i] == i)    //如果出现过并且是关系里最小的根节点
        {
            res.push_back({i, node[i].cnt, node[i].House / node[i].cnt, node[i].S / node[i].cnt});
        }
    }

    sort(res.begin(), res.end(), [&](Node a, Node b)  
         {
        if(a.S != b.S)return a.S > b.S;
        else return a.id < b.id; });
        //按照题目要求排序

    cout << res.size() << endl;
    for (auto t : res)
    {
        //注意前面补0
        printf("%04d %d %.3lf %.3lf\n", t.id, t.cnt, t.House, t.S);
    }

    return 0;
}