2019 USP-ICMC A - Jumping Buildings

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;
}