单调栈的c++实现

单调栈

1 问题提出

现在要解决一个问题

给出一个数组，找出这个数组每个下标对应的数左侧第一个比它小的数

要求时间复杂度小于 $$n^2$$

2 初步思考

简单方法，暴力枚举

for (int i  =0;i < n;i ++) {
    for (int j = i - 1;j >= 0;j --) {
        if (nums[j] < nums[i]) {
         // 合理 
            cout......
            break;
        }
    }
}

很明显时间复杂度为不够

3 进行优化

对于每个下标前的所有数可以使用栈来存，因为后入先出

一个个取出和当前数比较就是第二层循环的过程

怎么优化这个步骤

思路：

这个栈里面的元素可以找出性质。

比如有两个下标 $$i和j，i<j$$ $$但是nums[i]>nums[j]$$

当在这两个下标的右侧有一个数，有没有可能比他小的第一个数$$是nums[i]而不是nums[j]$$

不可能，因为$$nums[j]$$更小，而且更接近

所以我们要删除栈当中如上的这种关系

用图来展示，更为明显

删除后的图就是单调递增的图，所以叫单调栈

4 实现代码

#include <iostream>
#include <cstdio>
#include <stack>

using namespace std;
const int N=100010;
int a[N],n;

int main() {
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    stack<int> s;
    for (int i = 0;i < n;i++) {
        while (s.size() && s.top() >= a[i]) s.pop(); // 比当前大的都弹掉
        if (s.size() == 0) {
            printf("%d ",-1);  // 没有比当前小的
        }else {
            printf("%d ",s.top());  // 输出比当前小的第一个
        }
        s.push(a[i]);
    }
    return 0;
}

复杂度可以认为是线性的，因为单调栈每次循环弹出这个过程可以看作是常数级别的时间复杂度