1114. Family Property (25) 并查集

题目地址

和前面的1107类似，只是更复杂一些。
使用一个结构体Mem来记录每个人名下的财产（es),地产（area），家庭成员数（num）；借助’std::set’的去重功能，使用’set<>’记录所有出现的人的ID;使用map<int,vector<int>>记录集合根节点对应的所有子节点。并将集合中所有人的平均信息存储到该集合中ID最小的人头上，然后将其存入vector<int> final，最后排序输出。

//
// Created by aleafall on 16-10-2.
//

#include <iostream>
#include <map>
#include <cstdio>
#include <vector>
#include <set>
#include <algorithm>

using namespace std;

const int maxn = 10001;

struct Mem {
    double es, area;
    int num;

    Mem() : num(1), es(0), area(0) {}
} mem[maxn];

int father[maxn];

bool cmp(int a, int b) {
    double avg_area1 = mem[a].area / mem[a].num;
    double avg_area2 = mem[b].area / mem[b].num;
    if (avg_area1 != avg_area2) return avg_area1 > avg_area2;
    return a < b;
}

int findFather(int x) {
    if (x == father[x]) return x;
    else {
        int F = findFather(father[x]);
        father[x] = F;
        return F;
    }
}

void Union(int a, int b) {
    int faA = findFather(a);
    int faB = findFather(b);
    if (faA != faB) father[faB] = faA;
}

int main() {
    map<int, vector<int>> mp;
    set<int> total;
    vector<int> final;
    for (int i = 0; i < maxn; ++i) {
        father[i] = i;
    }
    int n, k, id, fa, mo, kid;
    cin >> n;
    for (int i = 0; i < n; ++i) {
        scanf("%d%d%d%d", &id, &fa, &mo, &k);
        total.insert(id);
        if (fa != -1) {
            total.insert(fa);
            Union(id, fa);
        }
        if (mo != -1) {
            total.insert(mo);
            Union(id, mo);
        }
        for (int j = 0; j < k; ++j) {
            scanf("%d", &kid);
            total.insert(kid);
            Union(kid, id);
        }
        scanf("%lf %lf", &mem[id].es, &mem[id].area);
    }
    for (set<int>::const_iterator it = total.cbegin(); it != total.cend(); ++it) {
        int F = findFather(*it);
        mp[F].push_back(*it);
    }
    for (auto it = mp.cbegin(); it != mp.cend(); it++) {
        sort(mp[it->first].begin(), mp[it->first].end());
        if (mp[it->first].size() > 1) {
            for (int i = 0; i < mp[it->first].size(); ++i) {
                if (i) {
                    mem[mp[it->first][0]].es += mem[mp[it->first][i]].es;
                    mem[mp[it->first][0]].area += mem[mp[it->first][i]].area;
                    mem[mp[it->first][0]].num++;
                }
            }
        }
        final.push_back(mp[it->first][0]);
    }
    sort(final.begin(), final.end(), cmp);
    cout << final.size() << endl;
    for (int i = 0; i < final.size(); ++i) {
        int cnt = mem[final[i]].num;
        printf("%04d %d %.3f %.3f\n", final[i], cnt, mem[final[i]].es / cnt,
               mem[final[i]].area / cnt);
    }
    return 0;
}