leetcode 503. 下一个更大元素 II

题目

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

示例 1:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

1.暴力
最坏可能接近O(n2)

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var nextGreaterElements = function (nums) {
    const res = []
    const n = nums.length
    let max = nums[0]
    for (let i = 0; i < n; i++) {
        max = Math.max(nums[i], max)
    }
    for (let i = 0; i < n; i++) {
        if (nums[i] === max) {
            res[i] = -1
        } else {
            let j = i + 1
            while (j < n && nums[j] <= nums[i]) {
                j++
            }
            if (j === n) {
                j = 0
            }
            while (j < i && nums[j] <= nums[i]) {
                j++
            }
            res[i] = nums[j]
        }
    }
    return res
};

1. 单调栈

注意思考方式
本题实则是考察单调递减序列维护，和如果出现非单调递减的情况怎么处理
注意：本题需要二轮循环，因为本题是循环队列，所以 i < 2*n， 这是本题值得学习的地方。

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var nextGreaterElements = function (nums) {
    //    单调递减栈
    const stack = []
    const n = nums.length
    const res = new Array(n).fill(-1)
    for (let i = 0; i < n * 2; i++) {
        while (stack && nums[stack[stack.length - 1]] < nums[i % n]) {
            res[stack.pop()] = nums[i % n]
        }
        if (i < n) {
            stack.push(i)
        }
    }
    return res
};