单调栈第二天(还没写完)

news2024/11/17 23:42:39

503.下一个更大元素II

力扣题目链接(opens new window)

给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。

示例 1:

输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；数字 2 找不到下一个更大的数；第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

看到题目的第一想法

利用昨天的单调栈的思路来写

但是它是循环数组该怎么解决？

遍历完一次后再遍历一次，将栈中剩余的弹出

最后剩余的赋值为-1

成功做出来

看到代码随想录的想法

没有我这么麻烦，for循环范围放大为2*length，nums[i%length]取模，逻辑和之前相同

若当前比栈里的小则直接存入

若当前比栈里的大，则弹出，栈顶下标对应的第一个最大值为栈里的值

自己实现时出现的困难

把result全部赋值为-1 自己就不用处理-1了

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        //也是用单调栈来写？
        //for循环两次到末尾
        Stack<Integer> stack = new Stack<>();
        //需要遍历到当前下标的前一个位置？
        stack.push(0);
        int[] result = new int[nums.length];
        for(int i=1;i<nums.length;i++){
            if(nums[stack.peek()]>=nums[i]){
                stack.push(i);
            }else{
                while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    result[stack.peek()] = nums[i];
                    stack.pop();
                }
                //不要忘了把这个值也push进去
                stack.push(i);
            }
        }
        //此时stack中残留的元素:
        //1 nums[i]末尾的元素
        //2 比较大的几个元素
        if(!stack.isEmpty()){
        for(int i=0;i<nums.length;i++){            
            while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
                    //弹出来的下标对应result 的下标
                    result[stack.peek()] = nums[i];
                    stack.pop();
            }
        }
        }
        while(!stack.isEmpty()){
            result[stack.peek()] = -1;
            stack.pop();
        }
        return result;
    }
}