【总结】单调栈(寻找下一个更大元素)


前言

单调栈在算法中,能够将O(n^2)的时间复杂度优化到O(n),相对的空间复杂度会增加,因为需要动态维护一个栈。在此总结一下 单调栈 题目,题目来源:
496. 下一个更大元素 I
503. 下一个更大元素 II


一、496. 下一个更大元素 I题目总结

496. 下一个更大元素 I
该方法核心是寻找到数组nums2各个元素的下一个更大元素,其实现方法为使用单调栈方法,又分为逆序寻找和顺序寻找,在此总结一下:

1.1 逆序寻找

其核心思想是使得栈中的元素满足单调递增,当不满足条件时进行弹出,以保持此特性。个人理解为:对于数组某一元素num,我们将其与栈顶的元素peekNum进行比较:

细节说明:
维持栈的单调递增特性本质上是 对元素num右侧的所有元素在不打乱原有的顺序下选出升序排序序列。 例如数组{8,2,3,4,1,10},对于元素8,其右侧元素的升序排序序列为:2,3,4,10, 可以发现此序列中10比8大,故8所对应的下一个更大的元素为10;

代码实现:

		// 逆序寻找下一个更大的数字
        for (int i = nums2。length-1; i >=0; i--){
            int num = nums2[i];
            while (!stack.isEmpty() && num > stack.peek()){
                stack.pop();
            }
            map.put(num, stack.isEmpty() ? -1 : stack.peek());
            stack.push(num);
        }

1.2 顺序寻找

其核心思想是令数组中的元素依次入栈,当遍历到某个元素num,大于栈顶的元素peekNum时,说明此时peekNum的下一个更大元素为num,记录下即可。

代码实现:

        // 正序寻找下一个更大的数字
        for (int i = 0; i < nums2.length; i++){
            int num = nums2[i];
            while (! stack.isEmpty() && num > stack.peek()){
                map.put(stack.pop(), num);
            }
            stack.push(num);
        }ck.push(num);
        }

二、503. 下一个更大元素 II

503. 下一个更大元素 II
其实现方法与 1.2顺序寻找方法一致,唯一区别是要遍历数组两遍,因为不只是寻找某元素右侧的下一个更大元素,而是按照顺时针循环寻找下一个更大的元素。

代码实现:

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] res = new int[n];
        Arrays.fill(res, -1);
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < 2*n; i++){
            while (!stack.isEmpty() && nums[ i % n] > nums[stack.peek() % n]){
                int index = stack.pop();
                res[index] = nums[i%n];
            }
            stack.push(i%n);
        }
        return res;

    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值