并查集有点乱讲道理。
我写这篇博客顺便屡一下思路。
1遇到并查集要把数据和father分开,无脑写一个find和合并。
2如果是记录所有节点的统计信息,由于没法通过父亲找孩子,所以直接用一个bool表记录所有见过的节点。
3可以写成路径压缩形式,注意用find函数内压缩得到的最终不见得压缩了。
比如a-b-c,d-e-f. 这时候d和a合并,d成为a的father,这里a和孩子并没有更新(没有办法更新,因为没有从父亲到孩子的反向路径)。需要在最后bool遍历的时候每个人find一下。
4对于bool取同根数据,我现在只想到了map比较方便。(或者set?)整体来说还是比较复杂的。主要原因在于father数组的下标不连续。排序用一个vector的sort,因为我没有记怎么改map的排序。
5注意初始化。
6注意别慌,有时候建立很多struct会觉得路做错了,其实不然,即使复杂度过高,也会有多半的分数。
#include<bits/stdc++.h>
using namespace std;
int father[10000];
bool flag[10000]={};
struct data{int s=0;int a=0;
}d[10000];
struct ans{int s,a;int h=0;int id;
};
int find(int a){if(father[a]==a)return a;int f=find(father[a]);father[a]=f;return f;
}
void hb(int a,int b){int f1=find(a);int f2=find(b);if(f1<f2)father[f2]=f1;else father[f1]=f2;
}
bool cmp(ans a,ans b){if((double)a.a/a.h!=(double)b.a/b.h)return (double)a.a/a.h>(double)b.a/b.h;return a.id<b.id;
}
int main(){for(int i=0;i<10000;i++)father[i]=i;int n,temp,p1,p2,k,child,s,a;cin>>n;for(int i=0;i<n;i++){cin>>temp>>p1>>p2;flag[temp]=true;if(p1!=-1){flag[p1]=true;hb(temp,p1);}if(p2!=-1){flag[p2]=true;hb(temp,p2);}cin>>k;for(int j=0;j<k;j++){cin>>child;flag[child]=true;hb(temp,child);}cin>>d[temp].s>>d[temp].a;}map<int,ans>mp;for(int i=0;i<10000;i++){if(flag[i]==true){find(i);mp[father[i]].h++;mp[father[i]].s+=d[i].s;mp[father[i]].a+=d[i].a;mp[father[i]].id=father[i]; // printf("%04d\n",father[i]);}}vector<ans>q;for(auto it=mp.begin();it!=mp.end();it++){q.push_back(it->second);}sort(q.begin(),q.end(),cmp);printf("%d\n",q.size());for(int i=0;i<q.size();i++){printf("%04d %d %.3f %.3f\n",q[i].id,q[i].h,(double)q[i].s/q[i].h,(double)q[i].a/q[i].h);}return 0;
}