单调栈的理解与应用

1.问题背景

考虑一个无序不重复数组arrarrarr如下:

3521476

我们希望为每个元素找到其左侧第一个比之小的元素(的下标),也就是得到下面的数组resresres

元素(下标)-10-1-1344
对应值/3//144

一般的,我们会想到对每个位置iii ,遍历j=i−1...0j=i-1...0j=i1...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-1i1的位置,因此栈中的元素必然都小于等于arr[i−1]arr[i-1]arr[i1].如果arr[i]>arr[i−1]arr[i] > arr[i-1]arr[i]>arr[i1],那么应该有res[i]=i−1res[i] = i-1res[i]=i1,否则i−1i-1i1将会出栈,直到从栈中找到第一个恰好小于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的总执行次数不会超过数组长度。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值