单调栈(Monotonic Stack)

单调栈是一种和单调队列类似的数据结构。单调队列主要在 O ( n ) O(n) O(n)复杂度下解决滑动窗口问题,单调栈主要在 O ( n ) O(n) O(n)复杂度下解决Next Greater Element(NGE)问题(对序列中的每个元素找下一个/上一个比他大/小的元素)

基本思想

单调栈维护一个,表示“待确定NGE的元素”,然后遍历序列。当碰上一个新元素,越靠近栈顶的元素离新元素的位置越近。所以不断比较新元素与栈顶,如果新元素比栈顶大,则可断定新元素就是栈顶的NGE,于是弹出栈顶并继续比较。直到新元素不比栈顶大,再将新元素压入栈。显然,这样形成的栈是单调递减的。

案例

举个例子,寻找数列中第 i i i个元素之后第一个大于它的元素值(没有则返回-1):

数组结果说明
10 3 7 4 12初始化
10 3 7 4 1210(索引:0)- i = 0 i=0 i=0时,栈空,则10入栈
10 3 7 4 1210 3(索引:0 1)- - i = 1 i=1 i=1时,10大于3,则3入栈
10 3 7 4 1210 7(索引:0 2)- 7 - i = 2 i=2 i=2时,3小于7,则3出栈,结果数组中索引1位置标记为7,7入栈
10 3 7 4 1210 7 4(索引:0 2 3)- 7 - - i = 3 i=3 i=3时,7大于4,则4入栈
10 3 7 4 1212(索引:4)12 7 12 12 - i = 4 i=4 i=4时,12大于4,12大于7,12大于10,结果数组中索引0、2、3位置标记为12,12入栈
10 3 7 4 1212 7 12 12 -1由于在 i = 4 i=4 i=4之后没有大于12的元素,所以返回-1

其的核心代码为:

public class Template {
    public void template(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];
        Stack<Integer> s = new Stack<>();

        for (int i = 0; i < n; i++) {
            if (!s.isEmpty()) {
                while (!s.isEmpty() && nums[s.peek()] < nums[i]) {
                    ans[s.pop()] = nums[i];
                }
            }
            s.push(i);
        }
        while (!s.isEmpty()) ans[s.pop()] = -1;
        for (int a : ans) System.out.printf("%d ", a);
    }

    public static void main(String[] args) {
        Template template = new Template();
        template.template(new int[]{10, 3, 7, 4, 12});
    }
}

题目链接

  1. 2289. 使数组按非递减顺序排列

参考文献:

  1. 单调栈
  2. 【模板】单调栈,上面的题目和该题目返回的结果不一致,该题返回的时索引,上题返回的是数值
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值