1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

// @Title: 下一个更大元素 II (Next Greater Element II)
// @Author: 15816537946@163.com
// @Date: 2021-03-06 20:59:31
// @Runtime: 32 ms
// @Memory: 6.6 MB
func nextGreaterElements(nums []int) []int {
	n := len(nums)
	ans := make([]int, n)
	for i := range ans {
		ans[i] = -1
	}

	stack := []int{}
	for i := 0; i < n*2-1; i++ {
		for len(stack) > 0 && nums[stack[len(stack)-1]] < nums[i%n] {
			ans[stack[len(stack)-1]] = nums[i%n]
			stack = stack[:len(stack)-1]
		}

		stack = append(stack, i%n)
	}

	return ans
}