单调栈的理解与应用

1.问题背景

考虑一个无序不重复数组$arr$如下：

3	5	2	1	4	7	6

我们希望为每个元素找到其左侧第一个比之小的元素（的下标），也就是得到下面的数组$res$

元素（下标）	-1	0	-1	-1	3	4	4
对应值	/	3	/	/	1	4	4

一般的，我们会想到对每个位置$i$ ，遍历$j=i-1...0$,的位置，找出第一个满足$arr[j]<arr[i]$的下标$j$。

然而，考虑将一个最小值放在某个降序数组的首位，例如$[0,6,5,4,3,2,1]$，对每个位置$i$的搜索则可能达到$O(n)$的复杂度，导致整个算法达到$O(n^2)$.

2.单调栈

通过栈的帮助，我们可以优化上述问题的解法。具体方案是，以下表变量$i$从左向右遍历整个数组，同时准备一个栈$stk$。
对于每个位置$i$，我们都判断栈顶（下标）元素是否大于当前$arr[i]$,如果是，则出栈，直到栈空或者是栈顶元素小于当前$arr[i]$.

若栈空，则说明$arr[i]$是最小的，那么$res[i]=-1$; 反之栈顶元素$j$则是$i$左侧第一个小于$arr[i]$的元素，记录$res[i]=j$.在这之后，我们再将当前的$i$放入栈中。

我们用图示展示这个过程：

我们总是用$i(arr[i])$的格式标注，其中$i$是下标，括号中是对应的值。其中，深红色的是原始数组，其上方的栈画出了判断当前$arr[i]$执行判断+出栈操作结束后的栈的情况，下方浅色数组是结果数组，下方对应的栈则是压入当前下标$i$后的情况。

我们不难发现并证明，这个栈中的元素总是自顶向下递减的，因此成为单调栈。对于第$i$个待定的位置而言，由于已经计算完了$i-1$的位置，因此栈中的元素必然都小于等于$arr[i-1]$.如果$arr[i]>arr[i-1]$，那么应该有$res[i]=i-1$，否则$i-1$将会出栈，直到从栈中找到第一个恰好小于$arr[i]$的元素。
我们尝试用C++实现：

vector<int> leftMin(vector<int> &arr) {
    stack<int> stk;
    vector<int> res(arr.size());
    for (int i = 0; i < arr.size(); ++i) {
        while (!stk.empty() && arr[stk.top()] > arr[i]){
            stk.pop();
        }
        res[i] = stk.empty() ? -1 : stk.top();
        stk.push(i);
    }
    return res;
}

一般认为，这个算法的复杂度是$O(n)$级别。少数读者或许有疑问，会考虑到这是两层循环，但实际上考虑内部的while循环，由于每个元素最多入栈一次，因此也最多出栈一次，也就是即便说整体考虑，内部while的总执行次数不会超过数组长度。