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

 #include <cstdio>
 #include <set>
 #include <vector>
 #include <algorithm>
 using namespace std;
 const int N=1e4+;
 const int inf=1e8;
 int n;
 int f[N];
 set<int>st,tt;
 set<int>::iterator it;
 struct node{
     int ns=,S=;
 }peo[N];
 struct edge{
     int Min=inf,num=;
     double avgHouse=,avgS=;
     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=; 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!=- && m!=-){
         st.insert(d);
         st.insert(m);
         join(d, m);
         join(x, d);
     }else if(d!=-){
         st.insert(d);
         join(x, d);
     }else if(m!=-){
         st.insert(m);
         join(x, m);
     }
 }
 int main(){
     int x,d,m,k,c;
     init();
     scanf("%d",&n);
     for (int i=; i<n; i++) {
         scanf("%d%d%d",&x,&d,&m);
         checkjoin(x, d, m);
         scanf("%d",&k);
         for (int j=; 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=; i<sus.size(); i++) {
         printf("%04d %d %.3lf %.3lf\n",sus[i].Min,sus[i].num,sus[i].avgHouse,sus[i].avgS);
     }
     return ;
 }