【单调栈】 LeetCode - 503. 下一个更大元素 II

503. 下一个更大元素 II

题目描述

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

示例 1:

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

解题思路

单调栈

    public static int[] nextGreaterElements(int[] nums) {
        int length = nums.length;
        int res[] = new int[length];
        Arrays.fill(res,-1); //默认都为1
        Stack<Integer> stack = new Stack<>();
        //相当于把数组遍历两遍
        for (int i = 0; i < length * 2; i++) {
            //遍历数组的第index（index从0开始）个元素，因为数组会遍历
            //两遍，会导致数组越界异常，所以这里要对数组长度进行求余
            int index = i % length;
            //单调栈，他存储的是元素的下标，不是元素具体值，从栈顶
            //到栈底所对应的值是递增的（栈顶元素在数组中对应的值最小，
            //栈底元素对应的值最大），如果栈顶元素对应的值比nums[index]小，
            //说明栈顶元素对应的值遇到了右边第一个比他大的值，然后栈顶元素出栈，
            //让他对应的位置变为nums[index]，也就是他右边第一个比他大的值，
            //然后继续判断……
            while (!stack.isEmpty() && nums[stack.peek()] < nums[index]){
                res[stack.pop()] = nums[index];
            }
            //当前元素的下标入栈
            stack.push(index);
        }
        return res;
    }