1.问题背景
考虑一个无序不重复数组arrarrarr如下:
| 3 | 5 | 2 | 1 | 4 | 7 | 6 |
|---|
我们希望为每个元素找到其左侧第一个比之小的元素(的下标),也就是得到下面的数组resresres
| 元素(下标) | -1 | 0 | -1 | -1 | 3 | 4 | 4 |
|---|---|---|---|---|---|---|---|
| 对应值 | / | 3 | / | / | 1 | 4 | 4 |
一般的,我们会想到对每个位置iii ,遍历j=i−1...0j=i-1...0j=i−1...0,的位置,找出第一个满足arr[j]<arr[i]arr[j] < arr[i]arr[j]<arr[i]的下标jjj。
然而,考虑将一个最小值放在某个降序数组的首位,例如[0,6,5,4,3,2,1][0,6,5,4,3,2,1][0,6,5,4,3,2,1],对每个位置iii的搜索则可能达到O(n)O(n)O(n)的复杂度,导致整个算法达到O(n2)O(n^2)O(n2).
2.单调栈
通过栈的帮助,我们可以优化上述问题的解法。具体方案是,以下表变量iii从左向右遍历整个数组,同时准备一个栈stkstkstk。
对于每个位置iii,我们都判断栈顶(下标)元素是否大于当前arr[i]arr[i]arr[i],如果是,则出栈,直到栈空或者是栈顶元素小于当前arr[i]arr[i]arr[i].
若栈空,则说明arr[i]arr[i]arr[i]是最小的,那么res[i]=−1res[i] = -1res[i]=−1; 反之栈顶元素jjj则是iii左侧第一个小于arr[i]arr[i]arr[i]的元素,记录res[i]=jres[i] = jres[i]=j.在这之后,我们再将当前的iii放入栈中。
我们用图示展示这个过程:

我们总是用i(arr[i])i(arr[i])i(arr[i])的格式标注,其中iii是下标,括号中是对应的值。其中,深红色的是原始数组,其上方的栈画出了判断当前arr[i]arr[i]arr[i]执行判断+出栈操作结束后的栈的情况,下方浅色数组是结果数组,下方对应的栈则是压入当前下标iii后的情况。
我们不难发现并证明,这个栈中的元素总是自顶向下递减的,因此成为单调栈。对于第iii个待定的位置而言,由于已经计算完了i−1i-1i−1的位置,因此栈中的元素必然都小于等于arr[i−1]arr[i-1]arr[i−1].如果arr[i]>arr[i−1]arr[i] > arr[i-1]arr[i]>arr[i−1],那么应该有res[i]=i−1res[i] = i-1res[i]=i−1,否则i−1i-1i−1将会出栈,直到从栈中找到第一个恰好小于arr[i]arr[i]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)O(n)O(n)级别。少数读者或许有疑问,会考虑到这是两层循环,但实际上考虑内部的while循环,由于每个元素最多入栈一次,因此也最多出栈一次,也就是即便说整体考虑,内部while的总执行次数不会超过数组长度。

1538

被折叠的 条评论
为什么被折叠?



