并查集有点乱讲道理。
我写这篇博客顺便屡一下思路。
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;
}