题意:此题给定一些人,然后给出一些人的相关信息,求出总的家族有多少个,以及每个家族的最小id以及家族的人数,平均财产,平均面积;
思路:此处要输出家族中id最小的那个,在我们归并的时候,要把id大的子树并到id小的根上面去,这样就能保证树根是最小的id了,这里也是一个比较有技巧性的一个点,再有就是计算人的个数的时候要定义一个访问数组,用于计算家族中的人数用的
代码:
#include<iostream>
#include<algorithm>
using namespace std;
struct DATA {//每个人的信息
int id, m, area, fid, mid;
int cid[10];
}info[10000];
struct node {//根节点的人的信息
int id, people, flag;
double area, m;
}ans[10000];
int root[10000];
int n = 0, k = 0, cnt = 0;
bool vis[10000] = { false };
int findRoot(int s) {
while (root[s] >= 0)
s = root[s];
return s;
}
void Union(int a,int b) {
int id1 = findRoot(a);
int id2 = findRoot(b);
if (id1 < id2)root[id2] = id1;//始终坚持把根id大些的并到id小的根上面去,这样就能够保证根的id为整个集合中id最小的了
else if (id1 > id2)root[id1] = id2;
}
bool com(node a,node b) {
if (a.area != b.area)return a.area > b.area;
else return a.id < b.id;
}
int main() {
fill(root, root + 10000, -1);
cin >> n;
for (int i = 0; i < n; i++) {
scanf("%d %d %d %d\n",&info[i].id,&info[i].fid,&info[i].mid,&k);
vis[info[i].id] = true;
for (int j = 0; j < k; j++) {
scanf("%d",&info[i].cid[j]);
Union(info[i].id, info[i].cid[j]);
vis[info[i].cid[j]] = true;//置为1表示访问过,用于计算总的人数
}
scanf("%d%d", &info[i].m, &info[i].area);
if (info[i].fid != -1) {
vis[info[i].fid] = true;
Union(info[i].id,info[i].fid);
}
if (info[i].mid != -1) {
vis[info[i].mid] = true;
Union(info[i].id,info[i].mid);
}
}
for (int i = 0; i < n; i++) {
int root = findRoot(info[i].id);
ans[root].id = root;
ans[root].area += info[i].area;
ans[root].m += info[i].m;
ans[root].flag = 1;
}
for (int i = 0; i < 10000; i++) {
if (vis[i])ans[findRoot(i)].people++;
if (ans[i].flag == 1)cnt++;
}
for (int i = 0; i < 10000; i++) {
if (ans[i].flag == 1) {
ans[i].m = ans[i].m*1.0 / ans[i].people;
ans[i].area = ans[i].area*1.0 / ans[i].people;
}
}
sort(ans,ans+10000,com);
printf("%d\n",cnt);
for (int i = 0; i < cnt; i++) {
printf("%04d %d %.3lf %.3lf\n",ans[i].id,ans[i].people,ans[i].m,ans[i].area);
}
system("pause");
return 0;
}