天梯赛L2-家庭房产

L2-007 家庭房产 (25分)

#include<bits/stdc++.h>
using namespace std;
struct Data
{
    int id,fid,mid,num,area;//个人编号，父编号，母编号，房产数，房产面积
    int cid[10];//子编号
}data[1005];//个人
struct node
{
    int id,people;
    double num,area;
    bool flag=false;//证明家族存在
}ans[10000];//家族
int father[10000];//父
bool visit[10000];//标志数组
int find(int x)
{
    if(x==father[x])
        return x;
    else
        return father[x]=find(father[x]);
}
void join(int a,int b)
{//以小标号为祖
    int fa=find(a);
    int fb=find(b);
    if(fa>fb)
        father[fa]=fb;
    else if(fa<fb)
    {
        father[fb]=fa;
    }
}
int cmp(node a,node b)
{
    if(a.area!=b.area)
    {
        return a.area>b.area;
    }
    else
        return a.id<b.id;
}
int main()
{
    int n,k,cnt=0;
    scanf("%d",&n);
    for(int i=0;i<10000;i++)
    {
        father[i]=i;
    }
    for(int i=0;i<n;i++)
    {
   scanf("%d %d %d %d",&data[i].id,&data[i].fid,&data[i].mid,&k);
   visit[data[i].id]=true;
   if(data[i].fid!=-1)
   {
       visit[data[i].fid]=true;
       join(data[i].fid,data[i].id);
   }
   if(data[i].mid!=-1)
   {
       visit[data[i].mid]=true;
       join(data[i].mid,data[i].id);
   }
   for(int j=0;j<k;j++)
   {
       scanf("%d",&data[i].cid[j]);
       visit[data[i].cid[j]]=true;
       join(data[i].cid[j],data[i].id);
   }
   scanf("%d %d",&data[i].num,&data[i].area);

    }
    for(int i=0;i<n;i++)
    {
        int id=find(data[i].id);
        ans[id].id=id;
        ans[id].num+=data[i].num;
        ans[id].area+=data[i].area;
        ans[id].flag=true;
    }
    for(int i=0;i<10000;i++)
    {
        if(visit[i])
            ans[find(i)].people++;
        if(ans[i].flag)
            cnt++;
    }
    for(int i=0;i<10000;i++)
    {
        if(ans[i].flag)
        {
            ans[i].num=(double)(ans[i].num*1.0/ans[i].people);
            ans[i].area=(double)(ans[i].area*1.0/ans[i].people);

        }
    }
    sort(ans,ans+10000,cmp);
    printf("%d\n",cnt);
    for(int i=0;i<cnt;i++)
    {
        printf("%04d %d %.3lf %.3lf\n",ans[i].id,ans[i].people,ans[i].num,ans[i].area);

    }
}