【PTA】家庭房产

曾经终败给现在 2023-07-02 13:24 17阅读 0赞

#### 题目重述 ####

给定每个人的家庭成员和其自己名下的房产，请你统计出每个家庭的人口数、人均房产面积及房产套数。

#### 输入格式： ####

输入第一行给出一个正整数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

#### 题解 ####

使用并查集，将一个家庭的成员都放到一个集合中，最后求集合的相关信息。

如果不知道并查集，这篇博文写的不错：[并查集][Link 1]  
本篇在思路上参考了[\[patl2-007\]家庭房产][patl2-007]

在代码中有详细注释，希望对大家有所帮助

#### C++ AC ####

#include <iostream>
    #include <bits/stdc++.h>
    #define MAX 10010
    using namespace std;
    struct person//用于保存输入样例的结构体
    {
        
        int id;//编号
        int f_id;//父亲的编号
        int m_id;//母亲的编号
        int k;//孩子的数量
        int child[10];//保存孩子的数组
        int house_num;//房产套数
        int sum_area;//总面积
    };
    struct rs//用于输出结果的结构体--家庭集合
    {
        
        int id;//最小编号
        int person_num;//家庭人数
        double ave_house;//平均房产数量
        double ave_area;//平均房产面积
    };
    int p[MAX];//用于保存上级
    bool vis[MAX];//用于保存编号，因为我们是用打表的方式，所以需要看哪个编号是有效的
    
    
    bool cmp(rs x,rs y)//自定义结构体比较
    {
        
        if(x.ave_area==y.ave_area)
        {
        
            return x.id<y.id;
        }
    
        else
        {
        
            return x.ave_area>y.ave_area;
        }
    
    }
    int find1(int x)//并查集中的寻找最上级
    {
        
        if(x==p[x])
        {
        
            return x;
        }
        else
        {
        
            return p[x]=find1(p[x]);
        }
    }
    void unite(int x,int y)//并查集中的合并
    {
        
        x=find1(x);
        y=find1(y);
        if(x==y)
            return;
        else
        {
        
            p[x]=y;
        }
    
    }
    int main()
    {
        
        ios::sync_with_stdio(false);
        memset(vis,0,sizeof(vis));
    
        person ps[10005];//用于保存输入样例
        rs result[10005];//用于保存输出样例
        for(int i=0; i<10005; i++)
        {
        
            p[i]=i;//默认自己的上级就是自己
        }
        int n;
        cin>>n;
    
        for(int i=0; i<n; i++)
        {
        
            cin>>ps[i].id>>ps[i].f_id>>ps[i].m_id>>ps[i].k;
            vis[ps[i].id]=true;//设置该编号有效
            if(ps[i].f_id!=-1)
            {
        
                unite(ps[i].id,ps[i].f_id);//将编号和父亲的编号合并
                vis[ps[i].f_id]=true;//设置父亲的编号也有效
            }
            if(ps[i].m_id!=-1)
            {
        
                unite(ps[i].id,ps[i].m_id);//母亲的编号
                vis[ps[i].m_id]=true;//设置父亲的编号也有效
            }
    
            for(int j=0; j<ps[i].k; j++)
            {
        
                cin>>ps[i].child[j];
                unite(ps[i].id,ps[i].child[j]);//将孩子的编号也都放到这个集合里
                vis[ps[i].child[j]]=true;//孩子的编号也有效
    
            }
            cin>>ps[i].house_num>>ps[i].sum_area;
        }
    
    
        for(int i=0; i<n; i++)
        {
        
            int id=find1(ps[i].id);//寻找最上级
            result[id].ave_house+=ps[i].house_num;//最上级加上当前编号的房产套数
            result[id].ave_area+=ps[i].sum_area;//最上级加上当前编号的总面积
        }
        //用来记录是不是最上级
        bool record[100010];
        memset(record,0,sizeof(record));
        int k=0;
        for(int i=0; i<=9999; i++)
        {
        
            if(vis[i])
            {
        
                int t=find1(i);
                record[t]=true;
    
                if(!result[t].person_num)//保证输出结果result中的id是家庭中的最小id
                {
        
                    result[t].id=i;
                }
                result[t].person_num++;
            }
        }
        for(int i=0; i<10010; i++)
        {
        
            if(record[i])
            {
        
                result[i].ave_area=result[i].ave_area*1.0/result[i].person_num;
                result[i].ave_house= result[i].ave_house*1.0/result[i].person_num;
            }
    
        }
        //进行结构体排序
        sort(result,result+10000,cmp);
        int ans=0;
        for(int i=0; i<10000; i++)
        {
        
            if(result[i].person_num)
            {
        
                ans++;
            }
        }
        printf("%d\n",ans);
        for(int i=0; i<ans; i++)
        {
        
            if(result[i].person_num)
            {
        
                printf("%04d %d %.3f %.3f\n",result[i].id,result[i].person_num,result[i].ave_house,result[i].ave_area);
            }
        }
        return 0;
    }

[Link 1]: https://blog.csdn.net/niushuai666/article/details/6662911
[patl2-007]: https://www.cnblogs.com/elpsycongroo/p/8526107.html