天梯赛 L2 - 007 家庭房产 【并查集】

传送门
// 题意: 题意就不多说了. 直接开讲思路.

首先有一个家族关系, 那么肯定这个要用并查集维护, 然后家庭人数就是集合的size, 也比较好维护, 比较难做的点是维护集合中每一个点有房产的, 所以对于这种题, 就不要再用下标从1开始累加他们了， 直接用下标来代替他们的编号, 因为这个编号范围很小, 我们可以直接枚举所有在这个范围内的点, 这种题都是这种套路, 记好了! 然后就是标记出现过的点, 然后for一遍把所有集合都取出来即可. 并查集就是普通的并查集, 注意我这个写法有一个坑点就是编号可以为0, 我这个就会出bug, 所以要通过该集合的人数来判断…… 具体细节请看代码.

AC Code

const int maxn = 1e4+5;
int fa[maxn], r[maxn];
void init() {
    for (int i = 0 ; i <= maxn-4 ; i ++) {
        fa[i] = i;
        r[i] = 1;
    }
}
int Find(int x) {
    return fa[x] == x ? x : fa[x] = Find(fa[x]);
}
void Un(int x, int y) {
    int fx = Find(x);
    int fy = Find(y);
    if (fx == fy) return ;
    if (r[fx] < r[fy]) swap(fx, fy);
    fa[fy] = fx;
    r[fx] += r[fy];
}
map<int, int>m1;
int a[maxn], b[maxn];
struct node {
    int id, num; db w1, w2;
    bool operator < (const node& _) const {
        if ((1.0*w2)/num == (1.0*_.w2)/_.num) return id < _.id;
        return (1.0*w2)/num > (1.0*_.w2)/_.num;
    }
}e[maxn];
void solve()
{
    init();
    int n; cin >> n;
    int idx = 0;
    for (int i = 1 ; i <= n ; i ++) {
        int tmp, f1, f2, son;
        cin >> tmp >> f1 >> f2;
        m1[tmp] = m1[f1] = m1[f2] = 1;
        if (f1 != -1) Un(tmp, f1);
        if (f2 != -1) Un(tmp, f2);
        int k; cin >> k;
        for (int j = 1 ; j <= k ; j ++) {
            cin >> son; m1[son] = 1;
            if (son != -1) Un(tmp, son);
        }
        cin >> a[tmp] >> b[tmp];
    }
    for (int i = 0 ; i <= maxn - 4 ; i ++) {
        if (!m1[i]) continue;
        int idx = Find(i);
        if (!e[idx].num) e[idx].id = i; // 不要用id来判断, 因为有0的存在.
        else e[idx].id = min(e[idx].id, i);
        e[idx].num++;
        e[idx].w1 += a[i]; e[idx].w2 += b[i];
    }
    int k = 0;
    for (int i = 0 ; i <= maxn - 4 ; i ++) {
        if (!e[i].num) continue;
        e[k++] = e[i];
    }
    sort(e, e+k);
    printf("%d\n", k);
    for (int i = 0 ; i < k ; i ++) {
        printf("%04d %d %.3f %.3f\n", e[i].id, e[i].num, 1.0*e[i].w1/e[i].num, 1.0*e[i].w2/e[i].num);
    }
}