题目链接:https://www.patest.cn/contests/gplt/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
题解:并查集。用数组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;
}