leetcode 503 下一个更大元素II (单调栈,循环数组)*

题目:给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。

解法:

1.单调栈思想,维护一个单调不增的数组的下标i~k,当出现一个比单调栈顶更大的元素num,则出栈后j~k的元素的最近更大元素即为num

2.注意是循环数组,单调栈最后剩下的下标要继续与开头的n位数组相比较, 可以将原数组拉直(扩展遍历两次),进行取模判断

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        stack = []
        n = len(nums)
        # 初始化原来的数组为-1即找不到的时候的值
        res = [-1]*n
        for i in range(2*n-1):
            while stack and nums[stack[-1]] < nums[i%n]:
                res[stack.pop()] = nums[i%n]
            stack.append(i%n)
        return res

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值