Leetcode 503. 下一个更大元素 II，难度：Medium。
原题请戳这里。

题目描述

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

示例1：

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

注意: 输入数组的长度不会超过 10000。

思路分析

本题是典型的Next Greater Element问题，解决方法一般都是单调栈。
在本题中，单调栈中保存的是数组的下标i，遍历数组，对于当前的nums[i]，将栈中所有小于nums[i]的下标全部弹出，这些下标对应值的下一个更大元素即为nums[i]，随后，将i入栈。另外，在本题中处理循环数组，只需要对下标取模即可。

关于单调栈的3种题型，可看@labuladong的LeetCode题解

AC代码

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

总结

本题相似题目：

• 496. 下一个更大元素 I
• 556. 下一个更大元素 III

用到的技巧

• 如何处理循环数组：
  1. 把循环数组拉直，即复制数组前n-1个数，拼接在原数组后面，模拟出循环数组
  2. 使用取模运算，即把下标 i 映射到数组 nums 长度的 0 - N 内。

  int[] arr = {1,2,3,4,5};
  int n = arr.length, index = 0;
  while (true) {
  	print(arr[index % n]);
  	index++;
  }
  

本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情