L2-007 家庭房产

题目链接

L2-007 家庭房产

分析

并查集题目，fa[i] 存储编号 $i$ 的父亲，sn[i] 记录编号 $i$ 是否出现，rela[] 记录人之间的连接，fnum[] 存储房产套数，area[] 存储人均面积，fsize[] 存储家庭人数

对于每一个家庭的信息，在这个人和他的每一个家庭成员之间建立连接，并记录他的房产信息。读入完成后对每一条链接进行合并，合并时同样合并房产信息，并标记所有出现的编号。最后统计信息排序输出即可

代码实现

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

const int MAXN = 10010;

vector<pair<int, int>> rela;
bool sn[MAXN];
int fnum[MAXN], fsize[MAXN], area[MAXN], fa[MAXN];

void init()
{
    for (int i = 0; i < MAXN; ++i)
        fa[i] = i, fsize[i] = 1;
}

int find(int x)
{
    return fa[x] == x ? x : (fa[x] = find(fa[x]));
}

void merge(int x, int y)
{
    int fx = find(x), fy = find(y);
    if (fx != fy)
    {
        if (fx > fy)
            swap(fx, fy);

        fa[fy] = fx;
        fsize[fx] += fsize[fy];
        fnum[fx] += fnum[fy];
        area[fx] += area[fy];
    }
}

bool camp(int x, int y)
{
    if (area[x] * fsize[y] == area[y] * fsize[x])
        return x < y;
    return area[x] * fsize[y] > area[y] * fsize[x];
}

int main()
{
    init();
    int T, n, f, m, k, c;
    cin >> T;
    while (T--)
    {
        cin >> n >> f >> m;
        if (~f)
            rela.push_back({n, f});
        if (~m)
            rela.push_back({n, m});
        sn[n] = true;

        cin >> k;
        while (k--)
        {
            cin >> c;
            rela.push_back({n, c});
        }
        cin >> fnum[n] >> area[n];
    }

    for (auto r : rela)
    {
        sn[r.first] = sn[r.second] = true;
        merge(r.first, r.second);
    }

    vector<int> px;
    for (int i = 0; i < MAXN; ++i)
    {
        if (sn[i] && fa[i] == i)
            px.push_back(i);
    }
    sort(px.begin(), px.end(), camp);
    cout << px.size() << endl;
    for (auto i : px)
        printf("%04d %d %.3f %.3f\n", i, fsize[i], 1.0 * fnum[i] / fsize[i], 1.0 * area[i] / fsize[i]);

    return 0;
}