下一个更大元素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]
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· DeepSeek 开源周回顾「GitHub 热点速览」
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了