天梯赛L2-007. 家庭房产（并查集+排序）

 

L2-007. 家庭房产

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

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

分析：这是天梯赛里面关于并查集的较为复杂的一题。其实不难，主要是要记录一些信息。一个比较常规的思路是，把相关信息全部放在家族代表元里，而代表元按题意需要以编号最小的为准。然后进行排序输出。

ps：不了解并查集的here。并查集入门与代码模板

#include<stdio.h>
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
int parent[10000],n[10000],sq[10000],setn[10000],exist[10000];//
struct inf{
	int num;
	double pn;
	double ps;
};
bool cmp(struct inf a,struct inf b)
{
	if(a.ps!=b.ps)//如果人均房产数不相等，则大的排前面，否则按照代表元的字典序。
		return a.ps>b.ps;
	else
		return a.num<b.num;
}
int find(int x)
{
	int root=x;
	while(parent[root]!=root)
		root=parent[root];
	while(parent[x]!=root){
		int t=parent[x];
		parent[x]=root;
		x=t;
	}
	return root;
}
void unio(int a,int b)
{
	int ra=find(a);
	int rb=find(b);
	if(ra!=rb)
		if(ra<rb)//这里需要注意，序号小的作为代表元。我一开始好像在这边把序号大的信息累加到代表元上，后来才知道，有的元素可能会漏掉。毕竟有的人和代表元合并的时候可能还未输入他的房产信息。
			parent[rb]=ra;
		else
			parent[ra]=rb;
}
int main()
{
	//freopen("L2-007. 家庭房产.txt","r",stdin);
	int m,num=0,k=0;
	for(int i=0;i<10000;i++)
		parent[i]=i;
	cin>>m;
	for(int i=0;i<m;i++){
		int p,f,m,num,hn,s;
		cin>>p>>f>>m>>num;
		int cur[num];//先把孩子信息存下来，然后再合并，这样该人的房产信息才能放到代表元身上。
		for(int i=0;i<num;i++)
			cin>>cur[i];
		cin>>hn>>s;
		n[p]=hn;//房产数量
		sq[p]=s; //面积
		exist[p]=1;
		if(f!=-1){//不是-1才要合并
			unio(p,f);
			exist[f]=1;//标记该人存在	
		}
		if(m!=-1){
			unio(p,m);
			exist[m]=1;
		}
		for(int i=0;i<num;i++){
			unio(cur[i],p);
			exist[cur[i]]=1;
		}
	}
	for(int i=0;i<10000;i++)
		if(exist[i]){//对存在的人找他们的祖先，在对应序号的索引处值+1
			setn[find(i)]++;
			if(find(i)!=i){//对房产信息统计，除代表元以外的信息全部汇总
				n[find(i)]+=n[i];
				sq[find(i)]+=sq[i];		
			}
		}
	for(int i=0;i<10000;i++){
		if(setn[i])
			num++;//统计家庭数量
	}
	printf("%d\n",num);
	inf list[num];
	for(int i=0;i<10000;i++)
		if(setn[i])
			list[k++]={i,n[i]*1.0/setn[i],sq[i]*1.0/setn[i]};			
	sort(list,list+num,cmp);
	for(int i=0;i<num;i++)
		printf("%04d %d %.3lf %.3lf\n",list[i].num,setn[list[i].num],list[i].pn,list[i].ps);
	return 0;
}