L2-007 家庭房产

题目

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

输入格式：

输入第一行给出一个正整数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

解题思路

（1）合并家族成员：获取n行的信息，并对每一行中出现的编号使用merge(first_id, other_id)进行合并，组成一个家族，使用vis[id]记录出现的成员编号，处理n行信息后，可以得到总共有多少个家族，并且每个id都会对应一个唯一的家族id（即家族最小成员id）；
（2）获取家族的房产和面积的总量：遍历n行个人信息，使用find(curLine.first_id)可以得到家族id，ans[家族id].houseNum += curLineData.houseNum，ans[家族id].houseArea += curLineData.houseArea
（3）获取家族总人数：遍历MAXN个id，通过vis[i]，可以知道此编号是否在上述数据中出现过，若出现过，则对应的家族人数+1
（4）获取家族人均房产和人均面积：遍历MAXN个id，计算人均房产和面积
（5）排序
（6）输出结果

代码

#include <iostream>
#include <algorithm>
using namespace std;

const int maxn = 10010;
typedef struct Node{
	int id, person;
	double house, area;
	bool flag;
}node;
node mdata[maxn];	// 对应初始数据 
node ans[maxn];		// 对应结果 
bool vis[maxn];		// 成员id是否出现 
int father[maxn];	// 存储家族id 
int n;

int find(int x)
{
	if (father[x] == x) return x;
	else return father[x] = find(father[x]);
}

void merg(int x, int y)
{
	int fx = find(x), fy = find(y);
	if (fx == fy) return ;
	else if (fx < fy) father[fy] = fx;
	else if (fx > fy) father[fx] = fy;
}

void init()
{
	for (int i = 0; i < maxn; i++)
	{
		father[i] = i;
	}
	scanf("%d", &n);
	int fid, mid, k, cid;
	for (int i = 0; i < n; i++)
	{
		scanf("%d%d%d%d", &mdata[i].id, &fid, &mid, &k);
		if (fid != -1) merg(mdata[i].id, fid), vis[fid] = true;
		if (mid != -1) merg(mdata[i].id, mid), vis[mid] = true;
		for (int j = 0; j < k; j++)
		{
			scanf("%d", &cid);
			merg(mdata[i].id, cid);
			vis[cid] = true;
		}
		scanf("%lf%lf", &mdata[i].house, &mdata[i].area);
		vis[mdata[i].id] = true;
		mdata[i].flag = true;
	}
}

bool cmp(node x, node y)
{
	if (x.area != y.area) return x.area > y.area;
	else return x.id < y.id;
}

void fun()
{
	for (int i = 0; i < n; i++)		// 统计家族房产和面积 
	{
		if (mdata[i].flag)
		{
			int p = find(mdata[i].id);
			ans[p].id = p;
			ans[p].house += mdata[i].house;
			ans[p].area += mdata[i].area;
			ans[p].flag = true;
		}
	}
	for (int i = 0; i < maxn; i++)		// 统计家族人数 
	{
		if (vis[i])
		{
			ans[find(i)].person += 1;
		}
	}
	int cnt = 0;
	for (int i = 0; i < maxn; i++)		// 统计平均值 
	{
		if (ans[i].flag)
		{
			cnt++;
			ans[i].house = ans[i].house / ans[i].person;
			ans[i].area = ans[i].area / ans[i].person;
		}
	}
	sort(ans, ans + 10000, cmp);
	printf("%d\n", cnt);
	for (int i = 0; i < cnt; i++)
	{
		printf("%04d %d %.3lf %.3lf\n", ans[i].id, ans[i].person, 
		ans[i].house, ans[i].area);
	}
}

int main()
{
	init();
	fun();
	return 0;
}