L2-007. 家庭房产

L2-007. 家庭房产

题目描述

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

Iutput

输入第一行给出一个正整数N（<=1000），随后N行，每行按下列格式给出一个人的房产：
编号 父 母 k 孩子1 … 孩子k 房产套数 总面积

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

Output

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

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

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

Sample Input

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

Sample Output

3
8888 1 1.000 1000.000
0001 15 0.600 100.000
5551 4 0.750 100.000

---

思路：

原理上就用了个并查集，但是由于信息比较多，所以后面处理起来还是比较麻烦的，结构体按照最后输出的要求来定义，同理，有些结果不要求输出的信息也不用不存起来。定义了三个结构体数组也是一步一步的保存状态压缩后的结果。
信息量比较大的题目，仔细读题比较关键。

---

代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=10001;
int p[maxn];

struct Node
{
    int num,id;
    double ans1,ans2;   //ans1代表人均套数，ans2代表人均面积
};

bool cmp(Node x,Node y)
{
    if(x.ans2==y.ans2)
        return x.id<y.id;
    else return x.ans2>y.ans2;
}
//并查集部分
void init() //初始化，每个节点都是一棵树
{
    for(int i=0; i<maxn; i++)
        p[i]=i;
}

int find(int x) //找到x的父节点
{
    if(x==p[x]) //x是树的的根节点
        return x;
    else return p[x]=find(p[x]); //找到x的父节点之后，继续向上找
}
void unit(int x,int y)
{
    int a=find(x);
    int b=find(y);
    if(a!=b)    //如果x,y不在同一棵树上，将两者相连
    {
        if(a>b) //将点值较小的作为两棵树的根节点
            p[a]=b;
        else p[b]=a;
    }
}

int main()
{
    init(); //初始化

    int n,p1,p2,d,a1,a2,i,k,id;
    Node a[maxn],b[maxn],c[maxn];
    //a用来记录原始数据[id]，b用来记录并查集处理后的部分[id]，c用来保存结果[cnt]
    int vis[maxn],m[maxn];

    memset(vis,0,sizeof(vis));  //标记id复用情况
    memset(m,0,sizeof(m));


    scanf("%d",&n);
    //readdate;
    for(i=0; i<n; i++)
    {
        scanf("%d%d%d%d",&a[i].id,&p1,&p2,&k);
        vis[a[i].id]=1;
        if(p1!=-1)
        {
            unit(a[i].id,p1);
            vis[p1]=1;
        }
        if(p2!=-1)
        {
            unit(a[i].id,p2);
            vis[p2]=1;
        }

        while(k--)
        {
            scanf("%d",&d);
            unit(d,a[i].id);
            vis[d]=1;
        }

        scanf("%d%d",&a1,&a2);
        a[i].ans1=a1;
        a[i].ans2=a2;
    }

    for(i=0; i<n; i++)  //找出各个家庭的房子的总套数和总面积
    {
        id=find(a[i].id);
        b[id].id=id;
        b[id].ans1+=a[i].ans1;
        b[id].ans2+=a[i].ans2;
    }

    for(i=0; i<maxn; i++)
    {
        if(vis[i])  //找出各个家庭的总人口
            b[find(i)].num++;
    }
    int cnt=0;
    for(i=0; i<maxn; i++)
    {
        if(vis[i])
        {
            id=find(i);
            if(!m[id])
            {
                m[id]=1;
                double x=double(b[id].num);
                c[cnt].ans1=b[id].ans1*1.0/x;
                c[cnt].ans2=b[id].ans2*1.0/x;
                c[cnt].id=id;
                c[cnt++].num=int(x);
            }
        }
    }
    printf("%d\n",cnt);
    sort(c,c+cnt,cmp);
    for(i=0; i<cnt; i++)
    {
        printf("%04d %d %.3f %.3f\n",c[i].id,c[i].num,c[i].ans1,c[i].ans2);
    }
    return 0;
}