1114 Family Property (25 分)

这题用到了并查集的思想，我的代码参考了柳婼的代码。柳神代码链接
坑点如下
1，统计人数时不能只遍历输入的id，还要遍历与之相关的人，因此开vis数组，记录所有出现的人。
2，比较函数注意要写return（在这一点上卡了很久）
3，注意find函数写法，不建议使用递归，易错。

int find(int x){
   while(father[x]!=x){
       x=father[x];
   }
    return x;
}

整体代码如下

#include<bits/stdc++.h>
using namespace std;
struct Node{
    int id,fa,mo,snum,area;
}node[1010];
struct ANS{
    int id;
    double tnum,tarea;
    int people;
    double avg,avgs;
    bool flag;
}ans[10010];
int father[10010];
bool vis[10010];
int find(int x){
   while(father[x]!=x){
       x=father[x];
   }
    return x;
}
void Union(int a,int b){
    int fac=find(a);
    int fab=find(b);
    if(fac>fab)father[fac]=fab;
    else father[fab]=fac;
}
bool cmp(ANS a, ANS b){
    if(a.avg!=b.avg) return a.avg>b.avg;
    else return a.id<b.id;
}
int main(){
    for(int i=0;i<10010;i++)
        father[i]=i;
    int n,k,kid;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
     scanf("%d%d%d%d",&node[i].id,&node[i].fa,&node[i].mo,&k);
        vis[node[i].id]=true;
     if(node[i].fa!=-1){Union(node[i].id,node[i].fa);
                       vis[node[i].fa]=true;}
     if(node[i].mo!=-1){Union(node[i].id,node[i].mo);
                       vis[node[i].mo]=true;}
     for(int l=0;l<k;l++){
      scanf("%d",&kid);
      Union(node[i].id,kid);
      vis[kid]=true;
     }
     scanf("%d%d",&node[i].snum,&node[i].area);
    }
    int cnt=0;
    for(int i=0;i<n;i++){
    int id=find(node[i].id);
    ans[id].id=id;
    ans[id].tnum+=node[i].snum;
    ans[id].tarea+=node[i].area;
    ans[id].flag=true;
    }
    for(int i=0;i<10010;i++){
       if(vis[i])
        ans[find(i)].people++;     
    }
    for(int i=0;i<10010;i++){
        if(ans[i].flag==true){
            cnt++;
            ans[i].avg=(1.0*ans[i].tarea)/ans[i].people;
            ans[i].avgs=(1.0*ans[i].tnum)/ans[i].people;
        }
    }
    sort(ans,ans+10010,cmp);
    printf("%d\n",cnt);
    for(int i=0;i<cnt;i++){
     printf("%04d %d %.3f %.3f\n",ans[i].id,ans[i].people,ans[i].avgs,ans[i].avg);
    }
    return 0;
}

另一种解法
测试点二一直未通过，经反复查找后发现是double引起的数据溢出问题，应该先求出总和再求平均数。

#include<bits/stdc++.h>
using namespace std;
set<int>s;
int father[10010];
vector<int>isroot[10010];
int findfather(int x){
    while(x!=father[x])
    {x=father[x];}
    return x;
}
void Union(int a,int b){
    int fa=findfather(a);
    int fb=findfather(b);
    fa<fb?father[fb]=fa:father[fa]=fb;
}
struct Node{
    int id,fa,ma;
    vector<int>child;
    int set,area;
}node[10010];
struct Node1{
    int id,num;
    int tset,tarea;
    double avgset,avgarea;
}node1[10010];
bool cmp(const Node1 &a,const Node1 &b){
   return a.avgarea!=b.avgarea?a.avgarea>b.avgarea:a.id<b.id;
}
int main(){
int n,id,k,kid,cnt=0;
    cin>>n;
    for(int i=0;i<10010;i++)father[i]=i;
    for(int i=0;i<n;i++){
    scanf("%d",&id);
    s.insert(id);
    scanf("%d%d%d",&node[id].fa,&node[id].ma,&k);
    if(node[id].fa!=-1){Union(id,node[id].fa);
                       s.insert(node[id].fa);}
    if(node[id].ma!=-1){Union(id,node[id].ma);
                   s.insert(node[id].ma);    }
    for(int l=0;l<k;l++)
    {  scanf("%d",&kid);
     Union(id,kid);
     s.insert(kid);
    node[id].child.push_back(kid);}
    node[id].id=id;
    scanf("%d%d",&node[id].set,&node[id].area);
}
   for(int i=0;i<10010;i++)
   {if(s.find(i)!=s.end())
     isroot[findfather(i)].push_back(i);
   }
    for(int i=0;i<10010;i++)
  {  if(isroot[i].size()!=0&&s.find(i)!=s.end()){cnt++;
       node1[i].id=i;
       node1[i].num=isroot[i].size();
     for(int l=0;l<isroot[i].size();l++){
       node1[i].tset+=node[isroot[i][l]].set;
       node1[i].tarea+=node[isroot[i][l]].area;
     }
       node1[i].avgset=node1[i].tset*1.0/node1[i].num;
       node1[i].avgarea=node1[i].tarea*1.0/node1[i].num;
   }
    }
    printf("%d\n",cnt);
    sort(node1,node1+10010,cmp);
    for(int i=0;i<cnt;i++)
        printf("%04d %d %.3f %.3f\n",node1[i].id,node1[i].num,node1[i].avgset,node1[i].avgarea);
   return 0;
}