L2-007 家庭房产 并查集

L2-007 家庭房产 （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

 这道题有一点绕，要判断父母是否存在后再combine哦。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int inf=0x3f3f3f3f;
int father[10005],vis[10005],gs[10005],bh[10005];
struct famliy
{
	int id,fid,mid;
	int child[10],oc;
	int ts,s;
}p[10005];
struct Famliy
{
	int id,rs;
	double rjt,rjs;
}P[10005];
int find(int x)  
{  
	int r=x,i=x,j;
	while(father[r]!=r)
	r=father[r];
	while(father[i]!=i)
	j=father[i],father[i]=r,i=j;
	return r;
}
void combine(int a,int b) 
{  
    int ta=find(a);  
    int tb=find(b);  
  	if(ta!=tb)  
   	father[ta]=tb;  
}  
bool cmp(Famliy x,Famliy y)
{
	if(x.rjs!=y.rjs) 
		return x.rjs>y.rjs;
	return x.id<y.id;
}
int main()
{
	int n,i,j,k,l,wp;
	for(i=0;i<=9999;i++)
	father[i]=i;
	memset(vis,0,sizeof vis);
	memset(gs,0,sizeof gs);
	memset(bh,inf,sizeof bh);
	scanf("%d",&n);
	for(i=0;i<n;i++)
	{
		scanf("%d",&wp);
		p[wp].id=wp;vis[wp]=1;
		scanf("%04d%04d",&p[wp].fid,&p[wp].mid);		
		if(p[wp].fid!=-1)
		{
			vis[p[wp].fid]=1;
			combine(p[wp].id,p[wp].fid);
		}		
		if(p[wp].mid!=-1)
		{
			vis[p[wp].mid]=1;
			combine(p[wp].id,p[wp].mid);
		}
		if(p[wp].fid!=-1&&p[wp].mid!=-1)
		combine(p[wp].fid,p[wp].mid);
		scanf("%d",&k);
		p[wp].oc=0;
		for(j=0;j<k;j++)
		{
			scanf("%04d",&p[wp].child[j]);
			combine(p[wp].id,p[wp].child[j]);
			if(j)
			combine(p[wp].child[j],p[wp].child[j-1]);
			vis[p[wp].child[j]]=1;
		}		
		p[wp].oc=j;
		//printf("oc:%d\n",p[i].oc);
		scanf("%d%d",&p[wp].ts,&p[wp].s);	
	}
	int sum=0;
	for(i=0;i<=9999;i++)
	{
		if(vis[i])
		{
			//printf("%04d:%04d\n",i,find(i));
			int fa=find(i);
			if(fa!=i)
			{
				p[fa].ts+=p[i].ts;
				p[fa].s+=p[i].s;
			}
			else
			sum++;
			gs[fa]++;	
			if(i<=bh[fa])
			{
				bh[fa]=i;
				//printf("bh:%04d\n",bh[fa]);
			}
		}
	}
	int op=0;
	for(i=0;i<=9999;i++)
	{
		if(gs[i])
		{
			//printf("%04d %d\n",bh[i],gs[i]);
			P[op].id=bh[i];
			P[op].rs=gs[i];
			P[op].rjt=1.0*p[i].ts/gs[i];
			P[op].rjs=1.0*p[i].s/gs[i];
			op++;
		}
	}
	printf("%d\n",sum);
	sort(P,P+op,cmp);
	for(i=0;i<op;i++)
	printf("%04d %d %.3lf %.3lf\n",P[i].id,P[i].rs,P[i].rjt,P[i].rjs);
}