ACWing 830. 单调栈（C++）

题目原文

ACWing 830. 单调栈（原题链接）

题目描述
给定一个长度为 $N$ 的整数数列，输出每个数左边第一个比它小的数，如果不存在则输出 $-1$。

输入格式
第一行包含整数 $N$，表示数列长度。

第二行包含 $N$ 个整数，表示整数数列。

输出格式
共一行，包含 $N$ 个整数，其中第 $i$ 个数表示第 $i$ 个数的左边第一个比它小的数，如果不存在则输出 $-1$。

数据范围
$1\le N\le 10^5$
$1\le 数列中元素\le 10^9$

输入样例：

5
3 4 2 7 5

输出样例：

-1 3 -1 2 2

解题思路

朴素做法：双重循环遍历，对每一个数都遍历前面所有数，找其到左边第一个比它小的数，较慢。

优化做法：单调栈

可以发现，若 $i<j<k$ ，且 $a[i]>=a[j]$ ，则 $a[i]$ 一定不可能是 $a[k]$ 左边第一个比它小的数:
(i) $a[j]<a[k]$ ，显然 $a[i]$ 就不可能是 $a[k]$ 左边第一个比它小的数；
(ii) $a[j]>=a[k]$ ，则有 $a[i]>a[j]>=a[k]$ ，显然 $a[i]$ 也不满足条件。

我们可以用栈来储存可能是答案的数值，显然由上述结论可知，栈中元素肯定是单调递增的。接下来需要注意的是，入栈、出栈操作如何进行，并且如何在栈中寻找满足条件的数。

对于每一个数，我们检查栈顶，比较栈顶元素与当前数的大小，若栈顶元素小于当前数，则栈顶元素即为所求值，并将当前数入栈；若栈顶元素大于等于当前数，则出栈（因为不是当前所求的满足条件的数，且因为其大于等于当前数，所以对后面出现的数已不可能构成有效答案）直到栈顶元素小于当前数为止。

完整代码

使用数组模拟单调栈：

#include <iostream>

using namespace std;

const int N = 1e5 + 10;
int a[N],stk[N] = {-1}; //数组a用来存放数列，数组stk模拟单调栈
//单调栈初始存入-1，当不存在比某数小的数时按题意输出-1
int cnt = 0;

int main()
{
    int n;
    scanf("%d", &n);
    
    while (n -- )
    {
        int x;
        scanf("%d", &x);
        
        //当栈非空且栈顶元素大于等于当前数时出栈
        while (cnt && stk[cnt] >= x) cnt --;
        
        printf("%d ", stk[cnt]);
        stk[++ cnt] = x;
    }
    
    return 0;
}

直接使用STL中的stack作为单调栈:

#include <iostream>
#include <stack>

using namespace std;

const int N = 1e5 + 10;
int a[N];

int main()
{
    int n;
    scanf("%d", &n);
    
    
    stack<int> s;
    s.push(-1);
    
    while (n -- )
    {
        int x;
        scanf("%d", &x);
        
        while (!s.empty() && s.top() >= x) s.pop();
        
        printf("%d ", s.top());
        s.push(x);
    }
    
    return 0;
}