洛谷P5788[模板]单调栈

【模板】单调栈P5788

题目背景

模板题，无背景。

2019.12.12 更新数据，放宽时限，现在不再卡常了。

题目描述

给出项数为 $n$ 的整数数列 $a_{1 \dots n}$。

定义函数 $f(i)$ 代表数列中第 $i$ 个元素之后第一个大于 $a_i$ 的元素的下标，即 $f(i)=\min_{i<j\leq n, a_j > a_i} \{j\}$。若不存在，则 $f(i)=0$。

试求出 $f(1\dots n)$。

输入格式

第一行一个正整数 $n$。

第二行 $n$ 个正整数 $a_{1\dots n}$。

输出格式

一行 $n$ 个整数表示 $f(1), f(2), \dots, f(n)$ 的值。

样例 #1

样例输入 #1

5
1 4 2 3 5

样例输出 #1

2 5 4 5 0

提示

【数据规模与约定】

对于 $30\%$ 的数据，$n\leq 100$；

对于 $60\%$ 的数据，$n\leq 5 \times 10^3$ ；

对于 $100\%$ 的数据，$1 \le n\leq 3\times 10^6$，$1\leq a_i\leq 10^9$。

code

#include<bits/stdc++.h>
using namespace std;
//单调栈模板
int s[3000005],ans[3000005];
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
    }
    stack<int> st;
    for(int i=n;i>=1;i--){
        while(!st.empty()&&s[st.top()]<=s[i]){      
            st.pop();
        }
        if(st.empty()){
            ans[i]=0;
        }else{
            ans[i]=st.top();
        }
        st.push(i);
    }
    for(int i=1;i<=n;i++){
        printf("%d ",ans[i]);
    }
    return 0;
}

举例[1,4,2,3,5]

从后往前扫单调递减

i=5
栈空{}
ans[i]=0
入栈5

i=4
栈{5}
s[栈顶下标5]=5>s[i]=3
ans[i]=栈顶元素
入栈4

i=3
栈{4,5}
s[栈顶下标4]=3>s[i]=2
ans[i]=栈顶元素
入栈3

i=2
栈{3,4,5}
s[栈顶下标3]=2<s[i]=4
出栈3 栈{4,5}
s[栈顶下标4]=3<s[i]=4
出栈4 栈{5}
入栈2
ans[i]=栈顶元素

i=1
栈{2,5}
s[栈顶下标2]=4>s[i]=1
ans[i]=栈顶元素
入栈1
循环结束栈{1,2,5}