patL2-007 家庭房产

题目

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

输入格式：

输入第一行给出一个正整数N（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

讲道理，这就是并查集，但是我写的真的是太丑了…..而且费了好长时间
先没管房数和面积，直接并查集，然后把同一个家庭的房产和面积加起来，计算结果
用到了 set ，因为set无重复的值，而且第一个是最小的值

//并查集
#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <set>
#include <algorithm>
using namespace std;
#define maxn 10010
int f[maxn];
bool vis[maxn];
struct Ans
{
    set<int> pnum;
    int hnum,harea;
    Ans(){}
}ans[maxn];
struct Pri
{
    char s[5];
    int pnum,num;
    double hnum,harea;
    Pri(){}
}pri[1005];

bool cmp(Pri a,Pri b)
{
    if(a.harea == b.harea)
        return a.num < b.num;
    else
        return a.harea > b.harea;
}

int getfather(int i)
{
    if(f[i] == i) return i;
    return f[i] = getfather(f[i]);
}
int changeid(char s[6])
{
    int ans = 0;
    for(int i = 0; i < 4; i++)
        ans += (int)(s[i]-'0') * pow(10.0,(3-i)*1.0);
    return ans;
}
int Merge(int i,int j)
{
    int x = getfather(i),y = getfather(j);
    if(x != y)
    {
        f[x] = y;
        for(auto it : ans[x].pnum)
            ans[y].pnum.insert(it);
    }
}

int main()
{
    //freopen("D://in.txt","r",stdin);
    int n;
    for(int i = 0; i < maxn; i++)
        f[i] = i;
    scanf("%d",&n);
    while(n--)
    {
        int flag1 = 0,flag2 = 0,k,hnum,harea,d[8];
        char data[8][6];
        for(int i = 0; i <= 2; i++)
            scanf("%s",data[i]);
        scanf("%d",&k);
        for(int i = 3; i <3+k; i++)
            scanf("%s",data[i]);
        scanf("%d %d",&hnum,&harea);

        if(data[1][0] == '-') flag1 = 1;
        if(data[2][0] == '-') flag2 = 1;

        for(int i = 0; i < 3+k; i++)
        {
            d[i] = changeid(data[i]);
            if((i == 1 && flag1) ||(i == 2 && flag2)) continue;
            vis[d[i]] = 1;
        }

        for(int i = 0; i < k+3; i++)
        {
            if((i == 1 && flag1) ||(i == 2 && flag2)) continue;
            ans[d[0]].pnum.insert(d[i]);
        }

        ans[d[0]].harea = harea;
        ans[d[0]].hnum = hnum;

        for(int i = 3; i < 3+k;i++)
            Merge(d[0],d[i]);

        int mark = 0;
        if(!flag1)
        {
            Merge(d[1],d[0]);
            if(!flag2)
            {
                mark = 1;
                Merge(d[1],d[2]);
            }
        }
        if(!mark && !flag2) Merge(d[0],d[2]);
    }
    int k = 0;
    for(int i = 0; i < 10000; i++)
       if(f[i] == i && vis[i])
    {
        int num = 1;
        int pl = 0;
        for(auto it : ans[i].pnum)
        {
            pl++;
            if(num == 1)
            {
                pri[k].num = it;
                pri[k].s[0] = it/1000 + '0';
                pri[k].s[1] = it/100%10 + '0';
                pri[k].s[2] = it/10%10 +'0';
                pri[k].s[3] = it%10 +'0';
            }
           if(it != i) ans[i].hnum += ans[it].hnum;
           if(it != i) ans[i].harea += ans[it].harea;
           num++;
        }
        pri[k].pnum = pl;
        pri[k].harea = (double)ans[i].harea/pl;
        pri[k++].hnum = (double)ans[i].hnum/pl;
    }
    sort(pri,pri+k,cmp);
    printf("%d\n",k);
    for(int i = 0; i < k ;i++)
        printf("%s %d %.3f %.3f\n",pri[i].s,pri[i].pnum,pri[i].hnum,pri[i].harea);

    return 0;
}