Question

503. 下一个更大元素 II

English Version

题目描述

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 _nums 中每个元素的 下一个更大元素_ 。

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

 

示例 1:

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

示例 2:

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

 

提示:

• 1 <= nums.length <= 104
• -109 <= nums[i] <= 109

解法

方法一：单调栈 + 循环数组

class Solution:
  def nextGreaterElements(self, nums: List[int]) -> List[int]:
    n = len(nums)
    ans = [-1] * n
    stk = []
    for i in range(n << 1):
      while stk and nums[stk[-1]] < nums[i % n]:
        ans[stk.pop()] = nums[i % n]
      stk.append(i % n)
    return ans

class Solution {
  public int[] nextGreaterElements(int[] nums) {
    int n = nums.length;
    int[] ans = new int[n];
    Arrays.fill(ans, -1);
    Deque<Integer> stk = new ArrayDeque<>();
    for (int i = 0; i < (n << 1); ++i) {
      while (!stk.isEmpty() && nums[stk.peek()] < nums[i % n]) {
        ans[stk.pop()] = nums[i % n];
      }
      stk.push(i % n);
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> nextGreaterElements(vector<int>& nums) {
    int n = nums.size();
    vector<int> ans(n, -1);
    stack<int> stk;
    for (int i = 0; i < (n << 1); ++i) {
      while (!stk.empty() && nums[stk.top()] < nums[i % n]) {
        ans[stk.top()] = nums[i % n];
        stk.pop();
      }
      stk.push(i % n);
    }
    return ans;
  }
};

func nextGreaterElements(nums []int) []int {
  n := len(nums)
  ans := make([]int, n)
  for i := range ans {
    ans[i] = -1
  }
  var stk []int
  for i := 0; i < (n << 1); i++ {
    for len(stk) > 0 && nums[stk[len(stk)-1]] < nums[i%n] {
      ans[stk[len(stk)-1]] = nums[i%n]
      stk = stk[:len(stk)-1]
    }
    stk = append(stk, i%n)
  }
  return ans
}

function nextGreaterElements(nums: number[]): number[] {
  const stack: number[] = [],
    len = nums.length;
  const res: number[] = new Array(len).fill(-1);
  for (let i = 0; i < 2 * len - 1; i++) {
    const j = i % len;
    while (stack.length !== 0 && nums[stack[stack.length - 1]] < nums[j]) {
      res[stack[stack.length - 1]] = nums[j];
      stack.pop();
    }
    stack.push(j);
  }
  return res;
}

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

方法二

class Solution:
  def nextGreaterElements(self, nums: List[int]) -> List[int]:
    n = len(nums)
    ans = [-1] * n
    stk = []
    for i in range(n * 2 - 1, -1, -1):
      i %= n
      while stk and stk[-1] <= nums[i]:
        stk.pop()
      if stk:
        ans[i] = stk[-1]
      stk.append(nums[i])
    return ans

class Solution {
  public int[] nextGreaterElements(int[] nums) {
    int n = nums.length;
    int[] ans = new int[n];
    Arrays.fill(ans, -1);
    Deque<Integer> stk = new ArrayDeque<>();
    for (int i = n * 2 - 1; i >= 0; --i) {
      int j = i % n;
      while (!stk.isEmpty() && stk.peek() <= nums[j]) {
        stk.pop();
      }
      if (!stk.isEmpty()) {
        ans[j] = stk.peek();
      }
      stk.push(nums[j]);
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> nextGreaterElements(vector<int>& nums) {
    int n = nums.size();
    vector<int> ans(n, -1);
    stack<int> stk;
    for (int i = n * 2 - 1; ~i; --i) {
      int j = i % n;
      while (!stk.empty() && stk.top() <= nums[j]) stk.pop();
      if (!stk.empty()) ans[j] = stk.top();
      stk.push(nums[j]);
    }
    return ans;
  }
};

func nextGreaterElements(nums []int) []int {
  n := len(nums)
  ans := make([]int, n)
  for i := range ans {
    ans[i] = -1
  }
  var stk []int
  for i := n*2 - 1; i >= 0; i-- {
    j := i % n
    for len(stk) > 0 && stk[len(stk)-1] <= nums[j] {
      stk = stk[:len(stk)-1]
    }
    if len(stk) > 0 {
      ans[j] = stk[len(stk)-1]
    }
    stk = append(stk, nums[j])
  }
  return ans
}

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