PAT 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

思路：并查集+父子关系数据更新

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 10005;
int vis[N]; //当前编号是否存在
int m[N];
int parent[N];
struct node
{
    double ans1,ans2;
    int id,num;
};

bool cmp(node a,node b){
	if(a.ans2!=b.ans2)
		return a.ans2>b.ans2;
	return a.id<b.id;
}
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
void make(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x!=y)
    {
        if(x>y)
            parent[x]=y;
        else
            parent[y]=x;
    }
}

int main()
{
    int n;
	node a[N],node b[N],mul[N];
	cin>>n;
	for(int i=0; i<=10000; i++)   //初始化
        parent[i]=i;
	memset(vis,0,sizeof(vis));
    memset(m,0,sizeof(m));
	    for(int i=0; i<n; i++) // 更新父子关系
    {
        int p1,p2,d,k;
        cin>>a[i].id>>p1>>p2;
        vis[a[i].id]=1;
        if(p1!=-1)
        {
            make(p1,a[i].id);
            vis[p1]=1;
        }
        if(p2!=-1)
        {
            make(p2,a[i].id);
            vis[p2]=1;
        }
        cin>>k;
        while(k--)
        {
            cin>>d;
            if(d!=-1)
            {
                make(a[i].id,d);
                vis[d]=1;
            }
        }
        cin>>a[i].ans1>>a[i].ans2;
    }
	for(int i=0;i<n;i++){             //求和家族
		int id=find(a[i].id);
		mul[id].id=id;
		mul[id].ans1+=a[i].ans1;
		mul[id].ans2+=a[i].ans2;
	}
	 for(int i=0; i<10000; i++)       //统计一个家族的人数
            if(vis[i])
			  mul[find(i)].num++;
	int cnt=0;
	for(int i=0;i<10000;i++){         //统计每个家族中的情况
		if(vis[i]){
			int id=find(i);
			if(!m[id]){
				m[id]=1;
                double x=double(mul[id].num);
                b[cnt].ans1=mul[id].ans1*1.0/x;
                b[cnt].ans2=mul[id].ans2*1.0/x;
                b[cnt].id=id;
                b[cnt++].num=int(x);
			}
		}
	}
	cout<<cnt<<endl;
	sort(b,b+cnt,cmp);
	for(int i=0;i<cnt;i++)
		printf("%04d %d %.3lf %.3lf\n",b[i].id,b[i].num,b[i].ans1,b[i].ans2);
    return 0;
}