题目链接:1114 Family Property (25分)
题意
给出N个人的 自己的id, 父亲的id,母亲的id,孩子个数,孩子id,房产数量,拥有房子总面积。求N个人一共有多少个家庭关系,输出每组家庭的 成员最小编号, 家庭人员个数,家庭平均房产数量,家庭平均土地面积,按照家庭平均土地面积递减排序,若相同则按照成员最小id递增排序。
分析
考察并查集。将每个人的id,父母id,孩子id并入集合,根为id最小的那个,且根值为集合个数的相反数。在设置一个家庭结构体,用来存放输出的答案。遍历完n个人后得到集合,且每组集合的根元素为集合中的id最小的。以每个集合根元素的id作为答案结构体的下标,然后在遍历每个人,找到所属集合根id,将资产情况存入家庭结构体中,然后在对家庭结构体按照题目要求排序。家庭个数为集合个数,也为排序后家庭总人数不为0的个数。
代码
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 10000;
typedef struct Person{
int id;
int father, mother;
vector<int> child;
int eNum;
double area;
Person() {
father = mother = -1;
eNum = 0;
area = 0;
}
}Person;
typedef struct Family{
int id, pNum;
int eNum;
double aNum, eavg, areaAvg;
}Family;
Person p[maxn];
Family f[maxn];
bool cmp(Family f1, Family f2) {
if (f1.areaAvg != f2.areaAvg)
return f1.areaAvg > f2.areaAvg;
return f1.id < f2.id;
}
int father[maxn];
void Union(int a, int b);
int findRoot(int x);
int main(int argc, char** argv) {
int n, id, cnum, k = 0;
scanf("%d", &n);
for (int i = 0; i < maxn; i++)
father[i] = -1;
for (int i = 0; i < n; i++) {
scanf("%d %d %d %d", &p[i].id, &p[i].father, &p[i].mother, &cnum);
if (p[i].father != -1)
Union(p[i].id, p[i].father);
if (p[i].mother != -1)
Union(p[i].id, p[i].mother);
for (int j = 0; j < cnum; j++) {
int cid;
scanf("%d", &cid);
Union(p[i].id, cid);
}
scanf("%d%lf", &p[i].eNum, &p[i].area);
}
for (int i = 0; i < n; i++) {
int root = findRoot(p[i].id);
f[root].id = root;
f[root].pNum = -father[root];
f[root].aNum += p[i].area;
f[root].eNum += p[i].eNum;
f[root].eavg = 1.0 * f[root].eNum / f[root].pNum; // 家庭平均地产数
f[root].areaAvg = f[root].aNum / f[root].pNum; // 家庭平均地面积
}
sort(f, f + maxn, cmp);
for (; f[k].pNum != 0; k++);
printf("%d\n", k);
for (int i = 0; i < k; i++) {
printf("%04d %d %.3f %.3f\n", f[i].id, f[i].pNum, f[i].eavg, f[i].areaAvg);
}
return 0;
}
int findRoot(int x) {
while (father[x] >= 0)
x = father[x];
return x;
}
// 将小值作为根,且根值为集合个数的负数
void Union(int a, int b) {
int fa = findRoot(a);
int fb = findRoot(b);
if (fa > fb) {
int t = father[fa];
father[fa] = fb;
father[fb] += t;
} else if (fa < fb) {
int t = father[fb];
father[fb] = fa;
father[fa] += t;
}
}