L2-007. 家庭房产(并查集)

#include <cstdio>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
const int N=1e4+5;
const int inf=1e8;
int n;
int f[N];
set<int>st,tt;
set<int>::iterator it;
struct node{
    int ns=0,S=0;
}peo[N];
struct edge{
    int Min=inf,num=0;
    double avgHouse=0,avgS=0;
    bool operator < (const edge x)const &{
        if(x.avgS==avgS){
            return x.Min>Min;
        }
        return x.avgS<avgS;
    }
}res[N];
vector<edge>sus;
void init(){
    for (int i=0; i<N; i++) {
        f[i]=i;
    }
}
int query(int x){
    int r=x;
    while (f[x]!=x) {
        x=f[x];
    }
    f[r]=x;
    return x;
}
void join(int x,int y){
    int fx=query(x);
    int fy=query(y);
    if(fx!=fy){
        f[fx]=fy;
    }
}
void checkjoin(int x,int d,int m){
    st.insert(x);
    if(d!=-1 && m!=-1){
        st.insert(d);
        st.insert(m);
        join(d, m);
        join(x, d);
    }else if(d!=-1){
        st.insert(d);
        join(x, d);
    }else if(m!=-1){
        st.insert(m);
        join(x, m);
    }
}
int main(){
    int x,d,m,k,c;
    init();
    scanf("%d",&n);
    for (int i=0; i<n; i++) {
        scanf("%d%d%d",&x,&d,&m);
        checkjoin(x, d, m);
        scanf("%d",&k);
        for (int j=0; j<k; j++) {
            scanf("%d",&c);
            st.insert(c);
            join(c, x);
        }
        scanf("%d%d",&peo[x].ns,&peo[x].S);
    }
    for (it=st.begin(); it!=st.end(); it++) {
        x=*it;
        int fx=query(x);
        if (x==fx)  tt.insert(x);
        res[fx].Min=min(res[fx].Min, x);
        res[fx].num++;
        res[fx].avgHouse+=peo[x].ns;
        res[fx].avgS+=peo[x].S;
    }
    for (it=tt.begin(); it!=tt.end(); it++) {
        x=*it;
        res[x].avgHouse/=res[x].num;
        res[x].avgS/=res[x].num;
        sus.push_back(res[x]);
    }
    sort(sus.begin(), sus.end());
    printf("%d\n",(int)sus.size());
    for (int i=0; i<sus.size(); i++) {
        printf("%04d %d %.3lf %.3lf\n",sus[i].Min,sus[i].num,sus[i].avgHouse,sus[i].avgS);
    }
    return 0;
}