503. 下一個更大元素 II

知識點：棧；單調

題目描述

給定一個迴圈陣列（最後一個元素的下一個元素是陣列的第一個元素），輸出每個元素的下一個更大元素。數字 x 的下一個更大的元素是按陣列遍歷順序，這個數字之後的第一個比它更大的數，這意味著你應該迴圈地搜尋它的下一個更大的數。如果不存在，則輸出 -1。

示例

輸入: [1,2,1]
輸出: [2,-1,2]
解釋: 第一個 1 的下一個更大的數是 2；
數字 2 找不到下一個更大的數； 
第二個 1 的下一個最大的數需要迴圈搜尋，結果也是 2。

解法一：單調

這道題和496題基本上是一樣的，都是採用單調棧的思想，也就是維持一個單調遞減棧，新來的要是比棧頂還小，那就可以入棧了，直到找到一個比棧頂大的元素，那證明這個元素就是棧裡某些元素的“下一個更大元素”。
這道題唯一的不同是要求是迴圈的，迴圈我們可以有兩種實現：

• 1.將這個陣列再複製一份接到原陣列後面，這當然可以，但是需要開闢新的空間；
• 2.經常用的是取餘數的運算。就可以將任意的數字對映到陣列長度0-N之間了。

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        Stack<Integer> stack = new Stack<>();  //存放下標；
        int n = nums.length;
        int[] res = new int[n];
        Arrays.fill(res, -1);  //初始化為-1；
        for(int i = 0; i < 2*n-1; i++){  //注意截至條件；
            while(!stack.isEmpty() && nums[i % n] > nums[stack.peek()]){
                res[stack.pop()] = nums[i % n];
            }
            stack.push(i % n);
        }
        return res;
    }
}
 

體會

• 會用單調棧，常見的觸發條件時“更大、更小”之類的字眼。
• 學會把陣列迴圈。