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

503.下一个更大元素II

题目描述

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

最优思路：单调栈

思路：

1. 将数组中所有元素全部置为 -1
2. 遍历两次，相当于循环遍历
3. 第一遍遍历，入栈索引： i
4. 只要后面元素比栈顶索引对应的元素大，索引出栈，更改result[stack.pop()]的数值
5. 最后栈里面剩余的索引对应的数组值，都为默认的 -1（因为后面未找到比它大的值）

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] result = new int[n];
        // 数组全部赋值为-1
        Arrays.fill(result, -1);

        // 求数组nums的n个数的每一个元素的一个更大元素，所以定义一个单调递增栈，栈中存放下标
        Stack<Integer> stack = new Stack<>();

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