L2-007 家庭房产（并查集&结构体）

1.题目

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

2.分析

        写这道题之前建议先把并查集的原理搞明白，然后另外的知识点就是结构体sort()的排序函数，其中题的坑点是编号为0000的人，可能会对一些判断语句产生影响导致测试点过不去。

3.代码

       

#include<iostream>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
const int MAX = 10000 + 10;

//定义结构体
struct P{
    double area;        //最好是double类型
    int people=1,uns;
}p[MAX];
int N, unmber, m, n, K, kid,houses, house;

int fa[MAX];   //并查集数组

set<int> s;
set<int> s1;
vector<int> v;

//结构体sort()比较函数
bool cmp(int a,int b){
    if(p[a].area*1.0/p[a].people==p[b].area*1.0/p[b].people) return a<b;
    return p[a].area*1.0/p[a].people>p[b].area*1.0/p[b].people;
}

//并查集的find()函数
int find(int x){
    if (x == fa[x]) return x;
    else{
        fa[x] = find(fa[x]);
        return fa[x];
    }
}

//并查集链接函数
void add(int a, int b)
{
    int f_a = find(a);
    int f_b = find(b);
    if (f_a > f_b) fa[f_a] = f_b;
    else fa[f_b] = f_a;
}
int main(){
    cin >> N;
    for(int i=0;i<10000;i++){fa[i]=i;}   //初始化并查集（避免不必要的判断（id为0000的人））
    for (int i = 0; i < N; i++){
        cin >> unmber >> m >> n >> K;
        s.insert(unmber);
        if (m >=0){
            s.insert(m);
            add(m, unmber);
        }
        if (n >=0){
            s.insert(n);
            add(n, unmber);
        }
        for (int j = 0; j < K; j++){
            cin >> kid;
            add(kid,unmber);
            s.insert(kid);
        }
        cin >> houses;
        p[unmber].uns+=houses;
        cin >> house;
        p[unmber].area+=house;
    }
    for (auto& it : s){
        find(it);          //简化并查集父结点的值（统一父结点）
        if(it!=fa[it]){
            p[fa[it]].people++;
            p[fa[it]].area+=p[it].area;
            p[fa[it]].uns+=p[it].uns;      //计算家庭房产和面积和人数
        }
        s1.insert(fa[it]);
    }
    cout<<s1.size()<<endl;
    int count=0;
    for (auto& it : s1){
        v.push_back(it);
        count++;
    }
    sort(v.begin(),v.end(),cmp);       //排序输出
    for(auto it:v){
        printf("%04d %d %.3f %.3f\n",fa[it],p[it].people,p[it].uns*1.0/p[it].people,p[it].area*1.0/p[it].people);
    }
    return 0;
}