|
503. 下一个更大元素 II
给定一个循环数组?nums?（?nums[nums.length - 1]?的下一个元素是?nums[0]?），返回?nums?中每个元素的?下一个更大元素?。
数字?x?的?下一个更大的元素?是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出?-1?。
示例 1:
输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数；
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
示例 2:
输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]
思路：
? 和每日温度几乎一模一样，得用单调栈来做。但这篇文章加了一点小难度，就是循环数组，就算是最后一个元素的下一个最大值，也又要从前面开始找。这点我们只需要将数组复制一倍，然后照常使用单调栈的模板，最终只取结果的一半就可以。关于单调栈的讲解可以直接看这里。
代码：
class Solution(object):
def nextGreaterElements(self, nums):
nums = nums*2 #将数组扩大两倍，这样就可以照常使用普通单调栈了
lenth =len(nums)
#单调栈模板
res = [0]*lenth
stack=[]
for i in range(lenth-1,-1,-1):
while(stack and stack[-1]<=nums[i]):
stack.pop()
res[i]=stack[-1] if stack else -1
stack.append(nums[i])
#最终拿单调栈前一半的结果即可
return res[:lenth/2]
|