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. 思路分析:
这道题目类似于之前力扣的496题,也是寻找下一个更大的数字,但是这里的数组可以看成是循环的,环形数组处理起来就比较麻烦,这里可以使用一个技巧是将原数组复制一遍接到原数组的后面,这样就可以将环形数组变为普通的一维数组,可以发现每一个数字后面都可以覆盖环形数组的长度为n的路径所以是符合要求的,并且求解出的答案也是满足要求的(例如例子[1,2,1][1,2,1]),这样我们就可以直接使用单调栈的模板了,这里需要注意的是只有在下标在n之内才更新答案(因为更新的才是原数组对应的答案数组)。
3. 代码如下:
from typing import List
class Solution:
def nextGreaterElements(self, nums: List[int]) -> List[int]:
n = len(nums)
# 将原数组复制一份接到后面
nums = nums + nums[:]
# 栈中存储的是元素下标
stack = list()
res = [-1] * n
for i in range(2 * n):
while stack and nums[stack[-1]] < nums[i]:
# 下标在n之内才更新答案
if stack[-1] < n:
res[stack[-1]] = nums[i]
stack.pop()
stack.append(i)
return res