(Leetcode 刷题) 下一个更大元素Ⅱ

题目描述

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

解法

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        //将默认值置为-1
        Arrays.fill(res, -1);
        LinkedList stack = new LinkedList<>(); 
        
        // i < 2 * nums.length 的范围很巧妙
        for(int i = 0; i < 2 * nums.length; i++) {
            while(!stack.isEmpty() && nums[stack.peek()] < nums[i % nums.length])
                res[stack.pop()] = nums[i % nums.length];
            stack.push(i % nums.length);
        }
        
        return res;
    }
}