题目要求:
给定每个人的家庭成员和其自己名下的房产,请你统计出每个家庭的人口数、人均房产面积及房产套数。
输入格式:
输入第一行给出一个正整数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
思路:
用并查集维护一个家族的成员,p[i]代表i属于哪家。为了方便后面的计算:
alnCnt[i]:i 所在家庭目前多少人。
house[i]:i 所在家庭目前几套房。
area[i]:i 所在家庭目前房产总面积套房。
vis[i]:表示有没有 i 这个人。
cnt:表示几个家庭。
结构体Family:id表示这家人中最小的编号,sum表示这家共有几人,a1表示平均房产套数,a2表示平均房产面积。
函数find():维护并查集,开始初始化p[]数组,p[i] = i,具体不做赘述。
函数join():合并两个家庭。
特别注意:有可能出现0000的节点,若测试点2、3、4没过,可能是这个原因。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 1e4 + 10;
int n, m, k, cnt;
int p[N], alnCnt[N], house[N], area[N];
bool vis[N];
struct Family{
int id, sum;
double a1, a2;
}ans[N];
bool cmp(Family x, Family y) //排序
{
if(x.a2 != y.a2) return x.a2 > y.a2; //人均房产面积不同,按人均面积降序
return x.id < y.id; //否则成员编号的升序
}
int find(int x)
{
if(p[x] != x) p[x] = find(p[x]);
return p[x];
}
void join(int a, int b)
{
if(b == -1) return; //如果输入的人是-1,表示过世
vis[b] = true;
int pa = find(a), pb = find(b); //分别找到两个人的祖宗节点,用于合并
if(pa != pb)
{ //题目要求输出家庭成员最小编号,因此将大的家庭附在小的家庭后。
//分别将节点、人口总数、房产套数、房产面积对应合并。
if(pa > pb) p[pa] = pb, alnCnt[pb] += alnCnt[pa], house[pb] += house[pa], area[pb] += area[pa];
else p[pb] = pa, alnCnt[pa] += alnCnt[pb], house[pa] += house[pb], area[pa] += area[pb];
}
}
int main(){
cin >> n;
for(int i = 0; i < N; i ++) p[i] = i, alnCnt[i] = 1;
for(int i = 0; i < n; i ++)
{
int id, a, b, k, t1, t2;
cin >> id >> a >> b >> k;
vis[id] = true; //表示有此人,用于后续筛选
join(id, a), join(id, b); //将这个人与父亲和母亲合并
while(k --)
{
int x;
cin >> x;
join(id, x); //将这个人和子女合并
}
cin >> t1 >> t2;
int pid = find(id);
house[pid] += t1, area[pid] += t2;
}
for(int i = 0; i <= 9999; i ++)
if(p[i] == i && vis[i]) //此人的祖宗是自己且此人存在,则说明他这一家编号最小的
ans[cnt ++] = {i, alnCnt[i], house[i] * 1.0 / alnCnt[i], area[i] * 1.0 / alnCnt[i]}; //将信息存入结构体
sort(ans, ans + cnt, cmp); //排序
cout << cnt << endl; //输出
for(int i = 0; i < cnt; i ++)
{
printf("%04d %d %.3lf %.3lf", ans[i].id, ans[i].sum, ans[i].a1, ans[i].a2);
if(i != cnt - 1) cout << endl; //回车处理
}
return 0;
}