A - Jumping Buildings
题目链接-Jumping Buildings
题目大意
给出n个数,在位置i上可移动到min(i+a[i],n)且只能移动一次,若中间有大于a[i]的数a[j],则只能移动到j-1,求在每个位置上能移动的距离
解题思路
单调栈,模拟即可,每次和把新输入的数字与栈顶比较,若大于栈顶就与栈中下一个数继续比较,直到找到栈顶大于输入的数字为止
附上代码
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f;
typedef long long ll;
typedef pair<int,int> PII;
const int N=1e5+5;
ll n,a[N],b[N],p;
int main(){
stack<ll> st;
cin>>n;
for(int i=1;i<=n;++i)
cin>>a[i];
a[n+1]=INF;
st.push(n+1);//n+1永不出栈
for(int i=n;i>=1;i--){
while(!st.empty()&&a[st.top()]<=a[i]){
st.pop();
}
ll t = st.top();
if(st.size()==1)
//如果没找到比它大的数,就在自身高度和到n的距离取min
b[i]=min(a[i],n-i);
else
//如果找到了就在第一种情况和到这个大于值前一栋楼距离取min
b[i] = min(t-i-1, min(a[i],n-i));
st.push(i);
}
for(int i=1;i<=n;i++){
cout<<b[i]<<" ";
} return 0;
}