开两个数组:to[i][j]表示从i这个位置向下的第2j个圆盘是哪个,f[i][j]表示流满从i这个位置向下的 2j 个圆盘需要多少体积的水。
详情见代码:
1 #include<bits/stdc++.h> 2 using namespace std; 3 const int N=1e5+10; 4 int d[N],c[N],r,v; 5 int n,q,f[N][20],to[N][20]; 6 int top,s[N]; 7 8 int read(){ 9 int x=0,f=1;char c=getchar(); 10 while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();} 11 while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar(); 12 return x*f; 13 } 14 15 int main(){ 16 n=read(),q=read(); 17 int sz=log2(n); 18 for(int i=1;i<=n;i++) d[i]=read(),c[i]=read(); 19 for(int i=n;i;i--){//用单调栈初始化出to[i][0],f[i][0] 20 while(top && d[i]>=d[s[top]]) top--; 21 to[i][0]=s[top],f[i][0]=c[i],s[++top]=i; 22 } 23 for(int j=1;j<=sz+1;j++)//ST的思想求出其他to和f 24 for(int i=1;i<=n;i++) 25 to[i][j]=to[to[i][j-1]][j-1],f[i][j]=f[i][j-1]+f[to[i][j-1]][j-1]; 26 while(q--){ 27 r=read(),v=read();//类似找LCA的做法 28 for(int j=sz+1;~j;j--) if(v>f[r][j]) v-=f[r][j],r=to[r][j]; 29 cout<<r<<endl; 30 } 31 }