503. 下一个更大元素 II

下一个更大元素 II

• 难度: medium
• 原始链接: https://leetcode.cn/problems/next-greater-element-ii
• 标签: 单调栈

解法一: 单调栈

go

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