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

题目链接：https://www.patest.cn/contests/gplt/L2-007

题目：

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

输入格式：

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

编号 父 母 k 孩子₁ ... 孩子_k 房产套数 总面积

其中 编号 是每个人独有的一个4位数的编号；父 和 母 分别是该编号对应的这个人的父母的编号（如果已经过世，则显示-1）；k（0<=k<=5）是该人的子女的个数；孩子_i是其子女的编号。

输出格式：

首先在第一行输出家庭个数（所有有亲属关系的人都属于同一个家庭）。随后按下列格式输出每个家庭的信息：

家庭成员的最小编号 家庭人口数 人均房产套数 人均房产面积

其中人均值要求保留小数点后3位。家庭信息首先按人均面积降序输出，若有并列，则按成员编号的升序输出。

输入样例：

106666 5551 5552 1 7777 1 1001234 5678 9012 1 0002 2 3008888 -1 -1 0 1 10002468 0001 0004 1 2222 1 5007777 6666 -1 0 2 3003721 -1 -1 1 2333 2 1509012 -1 -1 3 1236 1235 1234 1 1001235 5678 9012 0 1 502222 1236 2468 2 6661 6662 1 3002333 -1 3721 3 6661 6662 6663 1 100

输出样例：

38888 1 1.000 1000.0000001 15 0.600 100.0005551 4 0.750 100.000

题意：给出每个人的信息（父母，孩子，房产），统计家庭信息（人数，人均房产）。

题解：并查集。用数组vis记录每个人，最后遍VIS数组统计人数。

代码：

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<set>using namespace std;const int MAX=10000+10;const int NMAX=1000+10;struct personInfo{int id,fatid,momid,childCou;int childid[10];int houseCou,houseArea;personInfo(){};};personInfo person[NMAX];struct familyInfo{int id,num;int sumCou,sumArea;double cou,area;familyInfo(){};};familyInfo family[MAX];familyInfo ans[MAX];bool cmp(familyInfo a,familyInfo b){if(a.area==b.area) return a.id<b.id;return a.area>b.area;}int fa[MAX];int vis[MAX];int Find(int n){if(fa[n]==n) return n;return Find(fa[n]);}void Union(int u,int v){int fau=Find(u);int fav=Find(v);int fu=min(fau,fav);int fv=max(fau,fav);if(fv!=fu)fa[fv]=fu;}int main(){//freopen("in.txt","r",stdin);//freopen("out.txt","w",stdout);int N;while(scanf("%d",&N)!=EOF){for(int i=0;i<MAX;i++)fa[i]=i;memset(vis,0,sizeof(vis));for(int i=0;i<N;i++){scanf("%d%d%d%d",&person[i].id,&person[i].fatid,&person[i].momid,&person[i].childCou);vis[person[i].id]=1;if(person[i].fatid!=-1){vis[person[i].fatid]=1;Union(person[i].id,person[i].fatid);}if(person[i].momid!=-1) {vis[person[i].momid]=1;Union(person[i].id,person[i].momid);}for(int j=0;j<person[i].childCou;j++){scanf("%d",&person[i].childid[j]);vis[person[i].childid[j]]=1;Union(person[i].id,person[i].childid[j]);}scanf("%d%d",&person[i].houseCou,&person[i].houseArea);}set<int> root;root.clear();for(int i=0;i<N;i++){int id=Find(person[i].id);root.insert(id);family[id].id=id;family[id].sumCou+=person[i].houseCou;family[id].sumArea+=person[i].houseArea;}for(int i=0;i<MAX;i++){if(vis[i])family[Find(i)].num++;}printf("%d\n",(int)root.size());int t=0;for(set<int>::iterator it=root.begin();it!=root.end();it++){int id=*it;family[id].cou=1.0*family[id].sumCou/family[id].num;family[id].area=1.0*family[id].sumArea/family[id].num;ans[t++]=family[id];}sort(ans,ans+t,cmp);for(int i=0;i<t;i++){printf("%04d %d %.3lf %.3lf\n",ans[i].id,ans[i].num,ans[i].cou,ans[i].area);}}return 0;}