C++学习之路 | PTA(天梯赛)—— 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
這題一開始沒什么思路,感覺計算量很大,后來參考了別人的代碼,恍然大悟,剛開始看的時候可能有點蒙。
#include<iostream> #include<set> #include<vector> #include<map> #include<algorithm> using namespace std; const int maxn = 10000; int pre[maxn]; struct node {//存儲開始輸入的人的基本信息int id, house;double area; }; struct family {//輸出的家庭結構體int id, number;double house, area; }; bool cmp(family a, family b)//自定義比較函數 {if (a.area != b.area)return a.area > b.area;else return a.id < b.id; } int find(int x)//并查集模板,查找 {while (x != pre[x]) x = pre[x];return x; } void merge(int x, int y)//并查集模板,合并 {int a = find(x);int b = find(y);if (a != b) pre[a] = b; } int main() {int n, x;cin >> n;set<int>s;//存儲每人的編號,默認升序map<int, node>nodes;//記錄人的房產信息for (int i = 0; i < maxn; i++)pre[i] = i;//初始化pre數組for (int i = 0; i < n; i++){int id, p1, p2, k, house, id1;//分別記錄編號 父 母 k 孩子1 ... 孩子k 房產套數double area;// 總面積cin >> id >> p1 >> p2 >> k;s.insert(id);//插入idif (p1 != -1)//有爸爸{s.insert(p1);merge(id, p1);}if (p2 != -1)//有媽媽{s.insert(p2);merge(id, p2);}for (int j = 0; j < k; j++)//隨后的家庭成員{cin >> id1;s.insert(id1);merge(id1, id);}cin >> house >> area;nodes[id] = node{ id,house,area };//記錄該人的家庭信息}set<int>s1;//記錄家庭編號(也可以理解記錄著每個家庭的老大)map<int, vector<int>>m;//記錄家庭編號()和成員編號for (auto it = s.begin(); it != s.end(); it++){int fa = find(*it);m[fa].push_back(*it);s1.insert(fa);}cout << s1.size() << endl;//輸出家庭的個數vector<family>v1;for (auto it = s1.begin(); it != s1.end(); it++){int fa = *it;double house = 0, area = 0;for (auto tmp = 0; tmp < m[fa].size(); tmp++)//將每個家庭的成員及其房產計算匯總,放入family結構體中{house += nodes[m[fa][tmp]].house;area += nodes[m[fa][tmp]].area;}v1.push_back(family{ m[fa][0],int(m[fa].size()),house * 1.0 / (int)m[fa].size(),area * 1.0 / (int)m[fa].size() });}sort(v1.begin(), v1.end(), cmp);//進行排序for (int i = 0; i < v1.size(); i++) {printf("%04d %d %.3lf %.3lf", v1[i].id, v1[i].number, v1[i].house, v1[i].area);if (i < v1.size() - 1) puts("");} }總結
以上是生活随笔為你收集整理的C++学习之路 | PTA(天梯赛)—— L2-007 家庭房产 (25分)(带注释)(并查集)(精简)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 顶级隐写术,破坏一切解释器
- 下一篇: dnf时装预览怎么打开_DNF:史上最好