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

本文主要是介绍PTA 团体程序设计天梯赛-练习集 - L2-007 家庭房产（并查集，结构体），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

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

输入格式：
输入第一行给出一个正整数 $N （ \leq 1000 ）$ ，随后 $N$ 行，每行按下列格式给出一个人的房产：
编号父母 k 孩子1 ... 孩子k 房产套数总面积
其中编号是每个人独有的一个4位数的编号；父和母分别是该编号对应的这个人的父母的编号（如果已经过世，则显示-1）； $k （ 0 \leq k \leq 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){//注意前面补0printf("%04d %d %.3lf %.3lf\n", t.id, t.cnt, t.House, t.S);}return 0;
}