1.题目描述
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-greater-element-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.解题思路
维护一个递减栈,栈中存放的是原始列表的索引。
遍历原始列表,当前值比栈顶索引对应值小或者相等,当前索引i直接入栈。
当前值比栈顶索引对应值大,则弹出栈顶索引prevIndex
,该索引对应的ans值赋值为nums[i]。
但是注意到只遍历一次序列是不够的,例如序列 [2,3,1],最后单调栈中将剩余 [3,1],其中元素 [1]的下一个更大元素还是不知道的。
一个朴素的思想是,我们可以把这个循环数组「拉直」,即复制该序列的前 n-1个元素拼接在原序列的后面。这样我们就可以将这个新序列当作普通序列,用上文的方法来处理。
3.代码实现
class Solution:
def nextGreaterElements(self, nums: List[int]) -> List[int]:
index = [i for i in range(len(nums))] + [i for i in range(len(nums)-1)]
ans = [-1]*len(nums)
stack = []
for i in index:
while len(stack) > 0 and nums[i] > nums[stack[-1]]:
ans[stack[-1]] = nums[i]
stack.pop()
stack.append(i)
return ans