L2-007 家庭房产 (25 分)

L2-007 家庭房产 (25 分)

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

输入格式：
输入第一行给出一个正整数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

思路

并查集维护家族关系，同时维护家族人数和房屋面积和套数等各种信息
因为输入数据的住房在最后输入，可以把每个人作为他所在家族的顶梁柱，最后把他的住房信息加入即可

注意

更新家族人数或者房产信息的时候，需要先判断两者是不是已经在一个家族里面了，如果已经在同一家族就不要更新了

AC代码

#include <bits/stdc++.h>

using namespace std;

const int N = 10000;

int p[N];
int sz[N];
unordered_map<int, int> ys;
int house_cnt[N];
int house_area[N];
struct FamInfo
{
    double per_area;
    double per_cnt;
    int min_id;
    bool operator>(const FamInfo &f)const
    {
        if (per_area > f.per_area) return true;
        else if (per_area < f.per_area) return false;
        else 
        {
            return min_id < f.min_id;
        }
    }
};

int find(int x)
{
    if (x != p[x]) p[x] = find(p[x]);
    return p[x];
}


int main()
{
    int n;
    cin >> n;
    unordered_set<int> ids;
    for (int i = 0; i < N; i ++ ) p[i] = i, sz[i] = 1;
    for (int i = 0; i < n; i ++ )
    {
        int myid, faid, maid, cnt;
        cin >> myid >> faid >> maid >> cnt;
        ids.insert(myid);
        int pmy = find(myid);
        if (faid != -1) 
        {
            ids.insert(faid);
            int pfa = find(faid);
            if (pfa != pmy)
            {
                p[pfa] = pmy;
                sz[pmy] += sz[pfa];
                house_cnt[pmy] += house_cnt[pfa];
                house_area[pmy] += house_area[pfa];
            }
        }
        if (maid != -1)
        {
            ids.insert(maid);
            int pma = find(maid);
            if (pma != pmy)
            {
                p[pma] = pmy;
                sz[pmy] += sz[pma];
                house_cnt[pmy] += house_cnt[pma];
                house_area[pmy] += house_area[pma];
            }
        }
        while (cnt -- )
        {
            int kid;
            cin >> kid;
            ids.insert(kid);
            int pk = find(kid);
            if (pk != pmy)
            {
                p[pk] = pmy;
                sz[pmy] += sz[pk];
                house_cnt[pmy] += house_cnt[pk];
                house_area[pmy] += house_area[pk];
            }
        }
        int a, b;
        cin >> a >> b;
        house_cnt[pmy] += a;
        house_area[pmy] += b;
    }
    unordered_set<int> fam;
    for (auto id : ids)
    {
        int pi = find(id);
        fam.insert(pi);
    }
    cout << fam.size() << endl;
    vector<FamInfo> f;
    for (auto fid : fam)
    {
        int m_id = fid;
        double per_area = (double)house_area[fid] / sz[fid];
        double per_cnt = (double)house_cnt[fid] / sz[fid];
        for (int i = 0; i < N; i ++ )
            if (find(i) == fid)
            {
                m_id = i;
                break;
            }
        FamInfo it = {per_area, per_cnt, m_id};
        f.push_back(it);
    }
    sort(f.begin(), f.end(), greater<FamInfo>());
    for (int i = 0; i < f.size(); i ++ )
    {
        printf("%04d", f[i].min_id);
        cout  << ' ' << sz[find(f[i].min_id)] << ' ';
        printf("%.3lf %.3lf\n", f[i].per_cnt, f[i].per_area);
    }
    return 0;
}