Python :单调栈

 单调栈:用于求解一个序列中 每个元素 左边或右边第一个比其小的元素。

对应例题 : 百亿富翁


# 查找 「比当前元素大的元素」 就用 单调递增栈,查找「比当前元素小的元素」就用 单调递减栈。
# 从 「左侧」 查找就看 「插入栈」 时的栈顶元素,从 「右侧」 查找就看 「弹出栈」 时即将插入的元素

N = int(input())
building = list(map(int,input().split()))

def nextGreaterElement(nums) : # 寻找右边比这个元素大的元素
    res = []
    stack = []
    length = len(nums)
    for i in range(length-1,-1,-1) : # 倒序
        while stack and stack[-1] < nums[i] : # 将栈顶比当前元素小的元素全部pop出栈
            stack.pop()
        if len(stack) == 0 : # 若栈空了,则不存在满足的元素
            cur = -1
        else :
            cur = nums.index(stack[-1]) + 1 # 否则,找到比当前元素大的元素 即为当前的栈顶
        res.append(cur) # 将结果放入结果数组
        stack.append(nums[i]) # 将当前元素放入栈
    res.reverse()
    return res 

def beforeGreaterelement(nums) : # 寻找左边比这个元素大的元素 (原理与上相同)
    res = []
    stack = []
    length = len(nums)
    for i in range(length) :
        while stack and stack[-1] < nums[i] :
            stack.pop()
        if len(stack) == 0 :
            cur = -1
        else :
            cur = nums.index(stack[-1]) + 1
        res.append(cur)
        stack.append(nums[i])
    return res

print(beforeGreaterelement(building))
print(nextGreaterElement(building))
        

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UESTC_KS

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值