并查集 - 家庭房产

7-9 家庭房产（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

题意很明确，就是一个并查集就可以。就是写起来有些麻烦
代码示例 ：

#define ll long long
const int maxn = 1e6+5;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;

int n;
struct node
{
    int id;
    int f; // 表示父亲点
    int peo; // 总人数
    int num; // 房产套数
    int area; // 面积
}pre[10005];

struct node2
{
    int id;
    int peo;
    double an1, an2;    
}ans[10005];

int fid(int x){
    if (x != pre[x].f) pre[x].f = fid(pre[x].f);
    return pre[x].f;
}

void unit(int a, int b){
    int f1 = fid(a);
    int f2 = fid(b);
     
    if (f1 != f2) {
        if (f1 > f2) swap(f1, f2);
        pre[f2].f = f1;
    }
}
bool vis[10005];

bool cmp(node2 a, node2 b){
    if (a.an2 == b.an2) return a.id < b.id;
    else return a.an2 > b.an2;
}

int main() {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    int x, a, b;
    cin >> n;
    int k;
    for(int i = 0; i <= 10000; i++) pre[i].f = i;
    memset(vis, false, sizeof(vis));
    for(int i = 1; i <= n; i++){
        scanf("%d%d%d", &x, &a, &b);
        vis[x] = true; 
        if (a != -1) { unit(x, a); vis[a] = true; }       
        if (b != -1) { unit(x, b); vis[b] = true; }
        scanf("%d", &k);
        for(int i = 0; i < k; i++){
            scanf("%d", &a);
            unit(x, a);
            vis[a] = true;
        }
        scanf("%d%d", &a, &b);
        pre[x].num = a, pre[x].area = b;
    }
    for(int i = 0; i <= 10000; i++){
        if (vis[i]){ 
            int ff = fid(i);
            pre[i].f = ff;
            pre[ff].peo++;
            if (ff == i) continue; // 避免加重
            pre[ff].area += pre[i].area;
            pre[ff].num += pre[i].num;
        }
    }
    int cnt = 0;
    for(int i = 0; i <= 10000; i++){
        if (vis[i]){
            int ff = fid(i);
            if (ff == i) {
                ans[cnt].id = ff;
                ans[cnt].peo = pre[i].peo;
                ans[cnt].an1 = (1.0*pre[i].num)/pre[i].peo;
                ans[cnt].an2 = (1.0*pre[i].area)/pre[i].peo;
                cnt++;
            }
        }
    }
    sort(ans, ans+cnt, cmp);
    printf("%d\n", cnt);
    for(int i = 0; i < cnt; i++){
        printf("%.4d %d %.3f %.3f\n", ans[i].id, ans[i].peo, ans[i].an1, ans[i].an2);
    }
    return 0;
}
/*
2
2468 0001 0004 1 2222 1 500
2222 1236 2468 2 6661 6662 1 300

2
1234 5678 9012 1 0002 2 300
9012 -1 -1 3 1236 1235 1234 1 100
*/