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

把超级钢琴放到了树上。

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

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

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

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

 1 #include<bits/stdc++.h>
 2 #define pa pair<int,int>
 3 #define CLR(a,x) memset(a,x,sizeof(a))
 4 using namespace std;
 5 typedef long long ll;
 6 const int maxn=5e5+10;
 7 const ll inf=1e12;
 8 
 9 inline ll rd(){
10     ll x=0;char c=getchar();int neg=1;
11     while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
12     while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
13     return x*neg;
14 }
15 
16 struct Node{
17     int x,l,r,m;ll v;
18 };
19 bool operator < (Node a,Node b){return a.v<b.v;}
20 bool operator > (Node a,Node b){return a.v>b.v;}
21 int N,M,L,R;
22 int fa[maxn][22],mi[maxn][22],dep[maxn];
23 ll mn[maxn][22],v[maxn];
24 priority_queue<Node> q;
25 
26 inline ll getrmq(int x,int l,int r,int &m){
27     if(l<=0) return -inf;
28     ll vx=v[x];
29     for(int i=19;i>=0;i--){
30         if(fa[x][i]&&dep[fa[x][i]]>=l)
31             x=fa[x][i];
32     }
33     ll f=inf;
34     for(int i=19;i>=0;i--){
35         if(fa[x][i]&&dep[fa[x][i]]>=r){
36             if(mn[x][i]<f) f=mn[x][i],m=dep[mi[x][i]];
37             x=fa[x][i];
38         }
39     }
40     if(dep[x]>r&&mn[x][0]<f) f=mn[x][0],m=dep[mi[x][0]];
41     // printf("%d %d\n",vx,f);
42     return vx-f;
43 }
44 
45 int main(){
46     //freopen("","r",stdin);
47     int i,j,k;
48     N=rd();
49     for(i=2;i<=N+1;i++){
50         fa[i][0]=rd()+1;
51     }dep[1]=1;
52     for(i=2;i<=N+1;i++) v[i]=v[fa[i][0]]+rd(),dep[i]=dep[fa[i][0]]+1;
53     M=rd(),L=rd(),R=rd();
54 
55     for(i=1;i<=N+1;i++){
56         mn[i][0]=v[i],mi[i][0]=i;
57         for(j=0;fa[i][j]&&fa[fa[i][j]][j];j++){
58             fa[i][j+1]=fa[fa[i][j]][j];
59             if(mn[i][j]<mn[fa[i][j]][j]) mn[i][j+1]=mn[i][j],mi[i][j+1]=mi[i][j];
60             else mn[i][j+1]=mn[fa[i][j]][j],mi[i][j+1]=mi[fa[i][j]][j];
61         }
62     }
63     for(i=2;i<=N+1;i++){
64         Node p;
65         p.x=i,p.l=dep[i]-L,p.r=dep[i]-R-1;
66         // printf("%d %d %d\n",p.x,p.l,p.r);
67         if(p.l<=0) continue;
68         p.v=getrmq(p.x,p.l,p.r,p.m);
69         q.push(p);
70     }
71     ll ans=0;
72     for(i=1;i<=M;i++){
73         Node p=q.top();q.pop();
74         ans+=p.v;
75         if(p.m<p.l){
76             Node p1;
77             p1.x=p.x,p1.l=p.l,p1.r=p.m;
78             p1.v=getrmq(p1.x,p1.l,p1.r,p1.m);
79             q.push(p1);
80         }if(p.r<p.m-1){
81             Node p2;
82             p2.x=p.x,p2.l=p.m-1,p2.r=p.r;
83             p2.v=getrmq(p2.x,p2.l,p2.r,p2.m);
84             q.push(p2);
85         }
86     }
87     printf("%lld\n",ans);
88     return 0;
89 }

 

转载于:https://www.cnblogs.com/Ressed/p/9758745.html