题目:给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
解法:
1.单调栈思想,维护一个单调不增的数组的下标i~k,当出现一个比单调栈顶更大的元素num,则出栈后j~k的元素的最近更大元素即为num
2.注意是循环数组,单调栈最后剩下的下标要继续与开头的n位数组相比较, 可以将原数组拉直(扩展遍历两次),进行取模判断
class Solution:
def nextGreaterElements(self, nums: List[int]) -> List[int]:
stack = []
n = len(nums)
# 初始化原来的数组为-1即找不到的时候的值
res = [-1]*n
for i in range(2*n-1):
while stack and nums[stack[-1]] < nums[i%n]:
res[stack.pop()] = nums[i%n]
stack.append(i%n)
return res