A. Jumping Buildings
下标和a[i]混用,有点绕
#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define ll long long
#define PII piar
int a[100010],sta[100010],ans[100010];
signed main()
{
int n; cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
int tp=0;
for(int i=n;i>=1;i--){
while(tp&&a[i]>=a[sta[tp]]) tp--;
ans[i]=sta[tp];
sta[++tp]=i;
}
for(int i=1;i<=n;i++){
int an;
if(ans[i]==0) an=min(i+a[i],n);
else an=min(n,min(i+a[i],ans[i]-1));
cout<<an-i<<" ";
}
}
单调栈
参考:
序列 3,4,2,6,4,5,2,3:找出右边第一个比他大的值的下标
如果当前值比栈顶元素要大:弹栈
如果当前值比栈顶元素要小:压栈
->就是先把不合法的弹栈,然后必然要压栈。
//sta[]记录下标
for(int i=n;i>=1;i--)
{
while(tp&&a[sta[tp]]<=a[i]) tp--;
if(tp==0) ans[i]=0;
else ans[i]=sta[tp];
sta[++tp]=i;
}
从左边开始遍历,就是找左边第一个比它大的
题目