天体赛L2-007. 家庭房产

题目链接：https://www.patest.cn/contests/gplt/L2-007

这几天一直在练并查集。。。

#include <cstdio>
#include <algorithm>

using namespace std;

const double EPS=0.0001;

struct Node {
    int index,peo;
    double buil,area;

    bool operator < (const Node& a) const {
        return area-EPS>a.area||(abs(area-a.area)<EPS&&index<a.index);
    }
}que[10005];

struct Fam {
    int i,f,m,k[5],n,b,a;
}p[1005];

int n,pa,pb,tail;
int par[10005],mn[100005],peo[10005],buil[10005],area[10005];
bool vis[10005];

int getPar(int a) {
    if(par[a]!=a) {
        par[a]=getPar(par[a]);
    }
    return par[a];
}

void merg(int a,int b) {
    pa=getPar(a);
    pb=getPar(b);
    if(pa!=pb) {
        par[pb]=pa;
        mn[pa]=min(mn[pa],mn[pb]);
        peo[pa]+=peo[pb];
        buil[pa]+=buil[pb];
        area[pa]+=area[pb];
    }
}

int main() {
    while(1==scanf("%d",&n)) {
        for(int i=0;i<10000;++i) {
            par[i]=mn[i]=i;
            peo[i]=1;
            buil[i]=0;
            area[i]=0;
            vis[i]=true;
        }
        for(int i=0;i<n;++i) {
            scanf("%d%d%d%d",&p[i].i,&p[i].f,&p[i].m,&p[i].n);
            for(int j=0;j<p[i].n;++j) {
                scanf("%d",&p[i].k[j]);
            }
            scanf("%d%d",&buil[p[i].i],&area[p[i].i]);
        }
        for(int i=0;i<n;++i) {
            vis[p[i].i]=false;
            if(p[i].f!=-1) {
                merg(p[i].i,p[i].f);
                vis[p[i].f]=false;
            }
            if(p[i].m!=-1) {
                merg(p[i].i,p[i].m);
                vis[p[i].m]=false;
            }
            for(int j=0;j<p[i].n;++j) {
                if(p[i].k[j]!=-1) {
                    merg(p[i].i,p[i].k[j]);
                    vis[p[i].k[j]]=false;
                }
            }
        }
        tail=0;
        for(int i=0;i<10000;++i) {
            if(!vis[i]&&par[i]==i) {
                que[tail].index=mn[i];
                que[tail].peo=peo[i];
                que[tail].area=1.0*area[i]/peo[i];
                que[tail++].buil=1.0*buil[i]/peo[i];
            }
        }
        sort(que,que+tail);
        printf("%d
",tail);
        for(int i=0;i<tail;++i) {
            printf("%04d %d %.3lf %.3lf
",que[i].index,que[i].peo,que[i].buil,que[i].area);
        }
    }
    return 0;
}

参考链接：http://doc.okbase.net/idealism_xxm/archive/229098.html