503. 下一个更大元素 II

Posted 2022-04-11 hequnwang10

一、题目描述

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

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

示例 1：

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

示例 2：

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

二、解题

单调栈

这里使用单调栈来保存数组的下边，因为数组是循环的，所以可以将前n-1和数据复制到数组的后面，就是对下标取模。然后通过判断栈中下标对应的数值与当前值相比较，取下一个最大的值。

class Solution 
    public int[] nextGreaterElements(int[] nums) 
        int n = nums.length;
        int[] res = new int[n];
        Arrays.fill(res,-1);
        Deque<Integer> stack = new LinkedList<Integer>();
        for(int i = 0;i<2*n-1;i++)
            //将数组的前n-1个元素复制到数组的结尾，下标取模就可以了
            while(!stack.isEmpty() && nums[stack.peek()] < nums[i%n])
                res[stack.pop()] = nums[i%n];
            
            stack.push(i%n);
        
        return res;