7-11 家庭房产 (分数 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

思路：

用并查集维护一个家族的成员，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;
}