PTA 7-71 家庭房产并查集

最新推荐文章于 2024-04-09 20:08:11 发布

显卡燃烧中

最新推荐文章于 2024-04-09 20:08:11 发布

阅读量811

点赞数 2

文章标签：算法

本文链接：https://blog.csdn.net/weixin_73724647/article/details/130001112

版权

第一次做并查集，废了，不会，看大佬写的究极无敌霸气帅，觉得代码写得好哇，就学习了一下（相当于复刻），下面是原链接，感谢大佬，膜拜MMM(5条消息) 7-11 家庭房产（详细注释）_koala_cola的博客-CSDN博客

7-71 家庭房产

分数 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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

审题：

输入，处理，输出，算法的核心；

题目给这么多数据都是围绕一个核心那就是家庭，那显然是结构体或者类；

类之间的关系就是父母，儿子，这不就是树吗？（不过我们采取的是抽象的森林概念）

我们想是不是要把很多个类聚合为一个家族，计算平局面积和房产树，这就要用到核心算法：并查集了！

并查集主要操作：

初始化：

把每个点所在的集合初始化为自己；

查找：

查找元素所在的集合，即根节点；

合并：

将两个元素所在的集合合并为一个，就是换根节点，不过在合并前要记得判断两个集合是不是在同一个集合，在，就跳过了。

代码：

#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;


class people {
		public :
			int ID;
			int house_num, house_area, famaly;
			double aver_area()const {
				return ((double)(house_area) / (double)(famaly));
			}
			double aver_num()const {
				return ((double)(house_num) / (double)(famaly));
			}
			bool operator<(const people &b)const {
				if (aver_area() != b.aver_area()) {
					return aver_area() > b.aver_area();
				}
				else return ID < b.ID;
			}
}PEOPLE[10001]; 

int find(int x) {
	if (PEOPLE[x].ID != x) {
		PEOPLE[x].ID = find(PEOPLE[x].ID);
	}
	return PEOPLE[x].ID;
}

void swap(int *a, int *b) {
	 int temp = *a;
	 *a = *b;
	 *b = temp;
}

void union_f(int x, int y) {
	x = find(x);
	y = find(y);
	if (x == y) return;
	else if(x > y)
	    swap(&x, &y);
	PEOPLE[y].ID = x;
	PEOPLE[x].house_num += PEOPLE[y].house_num;
	PEOPLE[x].house_area += PEOPLE[y].house_area;
	PEOPLE[x].famaly += PEOPLE[y].famaly;
	PEOPLE[y].famaly = -1;
	PEOPLE[y].house_area = PEOPLE[x].house_area;//不能改，为什么/?????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????
}

bool visit[10002];
int temp[1005][10];

int main (void) {
	ios::sync_with_stdio(-1);
	cout.tie(NULL);
	cout.tie(NULL);
	
	int n;
	cin >> n;
	for(int i = 0; i < 10001; i++) {
		PEOPLE[i].ID = i;
		PEOPLE[i].house_num = 0;
		PEOPLE[i].house_area = 0;
		PEOPLE[i].famaly = 1;
	}
	for(int i = 0; i < n; i++) {
		cin >> temp[i][1] >> temp[i][2] >> temp[i][3] >> temp[i][4];
		visit[temp[i][1]] = true;
		if (temp[i][2] != -1) visit[temp[i][2]] = true;
		if (temp[i][3] != -1) visit[temp[i][3]]= true;
		for(int j = 0;  j < temp[i][4]; j++) {
			cin >> temp[i][5+j];
			visit[temp[i][5+j]] = true;
		} 
		cin >> PEOPLE[temp[i][1]].house_num >> PEOPLE[temp[i][1]].house_area; 
	}
	for(int i = 0; i < n; i++) {
		if(temp[i][2] != -1) 
			union_f(temp[i][1], temp[i][2]);
		if(temp[i][3] != -1) 
			union_f(temp[i][1], temp[i][3]);
		for(int j = 0; j < temp[i][4]; j++) {
			union_f(temp[i][1], temp[i][5 + j]);
		}
	}
	sort(PEOPLE, PEOPLE + 10000);
	
	int cnt = 0;
	while(visit[PEOPLE[cnt].ID])
		cnt++;
	cout << cnt << endl;
	for(int i = 0; i < cnt; i++) {
		printf("%04d %d %.3lf %.3lf\n", PEOPLE[i].ID, PEOPLE[i].famaly, PEOPLE[i].aver_num(), PEOPLE[i].aver_area());
	}
	
	return 0;
}