503. 下一个更大元素 II
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1] 输出: [2,-1,2] 解释: 第一个 1 的下一个更大的数是 2; 数字 2 找不到下一个更大的数; 第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
单调栈
思路
496. 下一个更大元素 I 同该题类似。最朴素的解决方法就是把数组翻倍,但是这样空间复杂度高了,可以根据取余操作来模仿数组翻倍。但不管怎么样,终究是要2n次循环。
题解
class Solution: def nextGreaterElements(self, nums: List[int]) -> List[int]: n = len(nums) res = [0 for _ in nums] stack = [] for i in range(2*n-1, -1, -1): while stack and nums[i%n] >= stack[-1]: stack.pop() res[i%n] = stack[-1] if stack else -1 stack.append(nums[i%n]) return res