首页 > 技术文章 > PTA家庭房产

alg-carlossxu 2020-11-26 21:49 原文

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

输入格式:
输入第一行给出一个正整数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 <bits/stdc++.h>
using namespace std;

const int N = 1e5+5;

struct person{
	int num;
	double ts, mj;
	person():num(1),ts(0.0),mj(0.0){};
};
int fa[N];
person p[N];
int vis[N];
vector<int> v;
int findfa(int x){
	return fa[x] == x ? x : fa[x] = findfa(fa[x]);
}
void join(int x, int y){
	int f1 = findfa(x);
	int f2 = findfa(y);
	if(f1 > f2) fa[f1] = f2;
	else if(f1 < f2) fa[f2] = f1;
}
bool cmp(int a, int b){
	return p[a].mj != p[b].mj ? p[a].mj > p[b].mj : a < b;
}

int main(){
	int n, id, k, f, m, son;
	cin >> n;
	for(int i = 0; i < N; i++) fa[i] = i;
	for(int i = 0; i < n; i++){
		cin >> id; 
		vis[id] = 1;
		scanf("%d %d", &f, &m);
		if(f!=-1){
			join(id,f);
			vis[f] = 1;
		} 
		if(m!=-1){
			join(id,m);
			vis[m] = 1;
		} 
		cin >> k;
		for(int j = 0; j < k; j++){
			scanf("%d", &son);
			join(id,son);
			vis[son] = 1;
		} 
		cin >> p[id].ts >> p[id].mj;
	}
	for(int i = 0; i < N; i++){
		if(vis[i]){
			int k = findfa(i);
			if(k == i) v.push_back(i);
			else{
				p[k].num++;
				p[k].ts += p[i].ts;
				p[k].mj += p[i].mj;
			}
		}
	}
	for(int i = 0; i < v.size(); i++){
		p[v[i]].ts /= p[v[i]].num;
		p[v[i]].mj /= p[v[i]].num;
	}
	sort(v.begin(), v.end(), cmp);
	printf("%d\n", v.size());
	for(int i = 0; i < v.size(); i++) 
		printf("%.4d %d %.3f %.3f\n",v[i],p[v[i]].num,p[v[i]].ts,p[v[i]].mj);
	return 0;
}

推荐阅读