P7167 [eJOI 2020 Day1] Fountain(单调栈+倍增)

P7167 [eJOI 2020 Day1] Fountain

题目描述:
对于一个喷泉,它有n层,每层有di的直径和ci的容量,对于m次查询,从index开始,总量有sum,问最后能流到哪一层,直径大的只能流向下一次直径比他大的或者水池.

思路:
先利用单调栈找到每一个圆盘能流向的下一个圆盘,通过倍增的想法存储当前圆盘流向的第二个圆盘,第四个,第八个…然后就能找到这一条流向的路,建一个二维数组保存倍增的区间和,然后对于每一次查询模拟即可

#include<bits/stdc++.h>
using namespace std;
#define int long long
int u[200005][25],f[200005][25],c[200005],d[200005];
signed main(){
	int n,q;
	scanf("%lld %lld",&n,&q);
	stack<int> sx;
	for(int i=1;i<=n;i++){
		scanf("%lld %lld",&d[i],&c[i]);
		while(!sx.empty()&&d[sx.top()]<d[i]){
			u[sx.top()][0]=i;
			sx.pop();			
		}
		sx.push(i);
	}
	for(int i=1;i<=n;i++)f[i][0]=c[i];
	for(int j=1;1<<(j-1)<=n;j++){
		for(int i=1;i<=n;i++){
			u[i][j]=u[u[i][j-1]][j-1];
			f[i][j]=f[i][j-1]+f[u[i][j-1]][j-1];
		}
	}
//	cout<<f[1][3]<<endl;
	int index,sum;
	while(q--){
		scanf("%lld %lld",&index,&sum);
		int lg=log2(n);
//		if(sum>f[index][lg+1])printf("0\n");
//		else {
			for(int j=lg+1;j>=0;j--){
				if(sum>f[index][j]){
					sum-=f[index][j];
					index=u[index][j];
				}
			}
			printf("%lld\n",index);
//		}
		
	}
	
	return 0;
}