L2-007. 家庭房产（并查集，变）

L2-007. 家庭房产

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个正整数N（<=1000），随后N行，每行按下列格式给出一个人的房产：

编号 父 母 k 孩子₁ ... 孩子_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

题解：

先把一个家庭的人用并查集模板连接起来。再计算房屋总数量，总面积，和根节点（因为需要输出最小编号，根节点存最小）。

再计算每个家庭有多少人，即计算有多少数根节点是同一个值，这里就要从0—MAX了，不是到n。

最后，计算平均套房数，面积，输出即可，这里还可以可以用一个结构体变量再存起来也可以直接输出。

#include <stdio.h>  
#include <string.h>  
#include <algorithm>  
#include <stack>  
#define inf 0x3f3f3f3f  
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
#define MAX 10005
struct node
{
	int id,ant;
	double num,s;
};
bool cmp(node x,node y)
{
	if(x.s==y.s)	
		return x.id<y.id;
	return x.s>y.s;
}
node fir[MAX];		//先后三个node结构体变量来存放 
node sec[MAX];
node last[MAX];
int vis[MAX];		//存在即vis=1 
int f[MAX];			
int flag[MAX];		//父节点标记为1，几个1就是几个家庭 
int find(int x)  
{
	if(x!=f[x])
		f[x]=find(f[x]);
	return f[x];
}
void join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		if(fx>fy)
			f[fx]=fy;
		else
			f[fy]=fx;
	}
}
int main()
{
	CLR(vis,0);
	CLR(flag,0);
	int T;
	scanf("%d",&T);
	for(int i=0;i<MAX;i++)
		f[i]=i;
	for(int i=0;i<T;i++)
	{
		int a,b,c,n;
		scanf("%d%d%d%d",&a,&b,&c,&n);
		vis[a]=1;
		if(b!=-1)	{join(a,b);vis[b]=1;}
		if(c!=-1) 	{join(a,c);vis[c]=1;}
		for(int j=1;j<=n;j++)
		{
			int ch;
			scanf("%d",&ch);
			if(ch!=-1)
			{
				join(a,ch);
				vis[ch]=1;
			}
		} 
		scanf("%lf%lf",&fir[i].num,&fir[i].s);
		fir[i].id=a;
	}
	//printf("%d\n",find(6666));
	for(int i=0;i<T;i++)
	{
		int x=find(fir[i].id);
		sec[x].id=x;
		sec[x].s+=fir[i].s;			//总面积 
		sec[x].num+=fir[i].num;	 //套房总数 
	}
	for(int i=0;i<MAX;i++)
	{
		if(vis[i])
			sec[find(i)].ant++;		//总人口	
	}
	int ans=0;
	for(int i=0;i<MAX;i++)
	{
		if(vis[i]&&!flag[find(i)])
		{
			int t=find(i);
			flag[t]=1;
			last[ans].id=t;
			last[ans].ant=sec[t].ant;		//总人口 
			last[ans].num=sec[t].num/sec[t].ant; 			//平均套房数
			last[ans++].s=sec[t].s/sec[t].ant;		//平均面积 
		}
	}
	sort(last,last+ans,cmp);
	printf("%d\n",ans);
	for(int i=0;i<ans;i++)
		printf("%04d %d %.3lf %.3lf\n",last[i].id,last[i].ant,last[i].num,last[i].s);
    return 0;  
}