并查集
题目大意
给出每个人名下的房产和家庭关系,求出每个家庭的总人数即人均房产数和房产面积。第一行输出家庭个数(所有有亲属关系的人都属于同一个家庭)。随后按下列格式输出每个家庭的信息:家庭成员的最小编号 家庭人口数 人均房产套数 人均房产面积。其中人均值要求保留小数点后3位。家庭信息首先按人均面积降序输出;若有并列,则按成员编号的升序输出。
思路
用并查集。分别用两个结构体存储,一个用来接收输入的数据,一个用于保存答案。(一开始我只用了一个,导致后面自己都乱套了,写不下去。为什么要用两个呢?因为可以注意到输入和结果的域有所不同,输入是按照个人对应名下房产,结果是按照家庭对应任何家庭成员的所有房产,即存在一个合并的关系。如果用一个的话,合并后的结果要与之前的分开,这样非常容易混乱。而如果在合并的时候,将结果做相加等运算迁移到另一个结构体中,思路就很清晰。)
其实输入的数据property可以不用开到10000个,在结构体中加入id即可,这样可以节省空间。(因为property不涉及id之间的关系,父子关系保存在另外的数组中;也不需要排序。它的作用只是别人调用它来获取信息)
AC代码()
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include <cstdio>
#include <vector>
#include<algorithm>
using namespace std;
#define N 10000
struct root {
int id, numpeople;
int num, area;
}ans[N];
struct own {
int id, num, eara;
}property[N];
bool exist[N] = { false };//为了计算家庭人数
int father[N];
int findFather(int x) {
while (x != father[x])x = father[x];
return x;
}
void Union(int a, int b) {
int faA = findFather(a);
int faB = findFather(b);
if (faA > faB)father[faA] = faB;
if (faA < faB)father[faB] = faA;
}
bool cmp(root a, root b) {
if (a.numpeople == 0 || b.numpeople == 0)return a.numpeople > b.numpeople;
else if (a.area * 1.00 / a.numpeople == b.area * 1.00 / b.numpeople)
return a.id < b.id;
else return a.area * 1.00 / a.numpeople > b.area * 1.00 / b.numpeople;
}
void initial() {
for (int i = 0;i < N;i++) {
father[i] = i;
property[i].num = property[i].eara = 0;
ans[i].numpeople = ans[i].area = ans[i].num = 0;
}
}
int main() {
int n, i, j;
scanf("%d", &n);
initial();
int owner, fid, mid;
for (i = 0;i < n;i++) {
scanf("%d%d%d", &owner, &fid, &mid);
exist[owner] = true;
if (fid >= 0) {
exist[fid] = true;
Union(owner, fid);
}
if (mid >= 0) {
exist[mid] = true;
Union(owner, mid);
}
int numchild, child;
scanf("%d", &numchild);
for (j = 0;j < numchild;j++) {
scanf("%d", &child);
exist[child] = true;
Union(owner, child);
}
scanf("%d%d", &property[owner].num, &property[owner].eara);
}
for (i = 0;i < N;i++) {
if (exist[i]) {
int fa = findFather(i);
ans[fa].numpeople++;
ans[fa].id = fa;
ans[fa].num += property[i].num;
ans[fa].area += property[i].eara;
}
}
sort(ans, ans + N, cmp);
int numfamily = 0;
for (i = 0;i < N && ans[i].numpeople;i++)numfamily++;
printf("%d
", numfamily);
for (i = 0;i < numfamily;i++) {
printf("%04d %d %.3f %.3f
", ans[i].id, ans[i].numpeople, ans[i].num * 1.0 / ans[i].numpeople, ans[i].area * 1.0 / ans[i].numpeople);
}
return 0;
}