简单并查集
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=10001;
int p[maxn];
struct Node
{
int num,id;
double ans1,ans2; //ans1代表人均套数,ans2代表人均面积
};
bool cmp(Node x,Node y)
{
if(x.ans2==y.ans2)
return x.id<y.id;
else return x.ans2>y.ans2;
}
//并查集部分
void init() //初始化,每个节点都是一棵树
{
for(int i=0; i<maxn; i++)
p[i]=i;
}
int find(int x) //找到x的父节点
{
if(x==p[x]) //x是树的的根节点
return x;
else return p[x]=find(p[x]); //找到x的父节点之后,继续向上找
}
void unit(int x,int y)
{
int a=find(x);
int b=find(y);
if(a!=b) //如果x,y不在同一棵树上,将两者相连
{
if(a>b) //将点值较小的作为两棵树的根节点
p[a]=b;
else p[b]=a;
}
}
int main()
{
init(); //初始化
int n,p1,p2,d,a1,a2,i,k,id;
Node a[maxn],b[maxn],c[maxn];
//a用来记录原始数据[id],b用来记录并查集处理后的部分[id],c用来保存结果[cnt]
int vis[maxn],m[maxn];
memset(vis,0,sizeof(vis)); //标记id复用情况
memset(m,0,sizeof(m));
scanf("%d",&n);
//readdate;
for(i=0; i<n; i++)
{
scanf("%d%d%d%d",&a[i].id,&p1,&p2,&k);
vis[a[i].id]=1;
if(p1!=-1)
{
unit(a[i].id,p1);
vis[p1]=1;
}
if(p2!=-1)
{
unit(a[i].id,p2);
vis[p2]=1;
}
while(k--)
{
scanf("%d",&d);
unit(d,a[i].id);
vis[d]=1;
}
scanf("%d%d",&a1,&a2);
a[i].ans1=a1;
a[i].ans2=a2;
}
for(i=0; i<n; i++) //找出各个家庭的房子的总套数和总面积
{
id=find(a[i].id);
b[id].id=id;
b[id].ans1+=a[i].ans1;
b[id].ans2+=a[i].ans2;
}
for(i=0; i<maxn; i++)
{
if(vis[i]) //找出各个家庭的总人口
b[find(i)].num++;
}
int cnt=0;
for(i=0; i<maxn; i++)
{
if(vis[i])
{
id=find(i);
if(!m[id])
{
m[id]=1;
double x=double(b[id].num);
c[cnt].ans1=b[id].ans1*1.0/x;
c[cnt].ans2=b[id].ans2*1.0/x;
c[cnt].id=id;
c[cnt++].num=int(x);
}
}
}
printf("%d\n",cnt);
sort(c,c+cnt,cmp);
for(i=0; i<cnt; i++)
{
printf("%04d %d %.3f %.3f\n",c[i].id,c[i].num,c[i].ans1,c[i].ans2);
}
return 0;
}