bzoj4458 GTY的OJ (优先队列+倍增)

把超级钢琴放到了树上。

这次不用主席树了..本来以为会好写一点没想到细节更多（其实是树上细节多）

为了方便，对每个点把他的那个L，R区间转化成两个深度a，b，表示从[a,b)选一个最小的前缀和（到根的和）减掉

为了更加方便，编号变为2~N+1，然后把2连到1上，1作为一个假根，权值为0

然后倍增去找那个a和b，记一记最小值的位置，然后劈开再加回到优先队列里就行了

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 const int maxn=5e5+;
 const ll inf=1e12;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 struct Node{
     int x,l,r,m;ll v;
 };
 bool operator < (Node a,Node b){return a.v<b.v;}
 bool operator > (Node a,Node b){return a.v>b.v;}
 int N,M,L,R;
 int fa[maxn][],mi[maxn][],dep[maxn];
 ll mn[maxn][],v[maxn];
 priority_queue<Node> q;

 inline ll getrmq(int x,int l,int r,int &m){
     if(l<=) return -inf;
     ll vx=v[x];
     for(int i=;i>=;i--){
         if(fa[x][i]&&dep[fa[x][i]]>=l)
             x=fa[x][i];
     }
     ll f=inf;
     for(int i=;i>=;i--){
         if(fa[x][i]&&dep[fa[x][i]]>=r){
             if(mn[x][i]<f) f=mn[x][i],m=dep[mi[x][i]];
             x=fa[x][i];
         }
     }
     if(dep[x]>r&&mn[x][]<f) f=mn[x][],m=dep[mi[x][]];
     // printf("%d %d\n",vx,f);
     return vx-f;
 }

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     N=rd();
     for(i=;i<=N+;i++){
         fa[i][]=rd()+;
     }dep[]=;
     for(i=;i<=N+;i++) v[i]=v[fa[i][]]+rd(),dep[i]=dep[fa[i][]]+;
     M=rd(),L=rd(),R=rd();

     for(i=;i<=N+;i++){
         mn[i][]=v[i],mi[i][]=i;
         for(j=;fa[i][j]&&fa[fa[i][j]][j];j++){
             fa[i][j+]=fa[fa[i][j]][j];
             if(mn[i][j]<mn[fa[i][j]][j]) mn[i][j+]=mn[i][j],mi[i][j+]=mi[i][j];
             else mn[i][j+]=mn[fa[i][j]][j],mi[i][j+]=mi[fa[i][j]][j];
         }
     }
     for(i=;i<=N+;i++){
         Node p;
         p.x=i,p.l=dep[i]-L,p.r=dep[i]-R-;
         // printf("%d %d %d\n",p.x,p.l,p.r);
         if(p.l<=) continue;
         p.v=getrmq(p.x,p.l,p.r,p.m);
         q.push(p);
     }
     ll ans=;
     for(i=;i<=M;i++){
         Node p=q.top();q.pop();
         ans+=p.v;
         if(p.m<p.l){
             Node p1;
             p1.x=p.x,p1.l=p.l,p1.r=p.m;
             p1.v=getrmq(p1.x,p1.l,p1.r,p1.m);
             q.push(p1);
         }if(p.r<p.m-){
             Node p2;
             p2.x=p.x,p2.l=p.m-,p2.r=p.r;
             p2.v=getrmq(p2.x,p2.l,p2.r,p2.m);
             q.push(p2);
         }
     }
     printf("%lld\n",ans);
     return ;
 }