最小栈的实现(Python)

一、题目描述
实现一个栈,该栈带有出栈(pop)、入栈(push)、取最小元素(get_min)三个方法,要保证这三个方法的时间复杂度都是O(1)。

二、解题思路
可能大部分在看到这个题后,第一反应是创建一个整型变量min,用来存储栈中的最小元素,每新进来一个元素,就让它和最小元素比较,只要它比最小元素小,就更新它为最小元素。当调用get_min时,直接返回min的值即可。

以上算法错误之处在于只考虑了进栈的情况,当栈顶元素出栈时,此时栈中的最小值我们并不能知道是多少。

所以,只存一个最小值是不够的,我们需要把栈中的历史最小值都存储下来。

详细步骤如下:
1、设原来的栈为A,我们新建的存储历史最小值的栈称之为B。
2、当第一个元素进入A时,此时A的最小值就是该元素,所以把该元素也放入栈B中。
3、每当新进来一个元素,都要比较新元素和栈A当前最小值的大小,如果新元素小于栈A当前最小值,则让新元素也进入栈B中。以此办法让栈B的栈顶元素始终是栈A的最小值。
4、当栈A有元素出栈时,如果出栈元素是栈A当前最小值,此时让栈B的栈顶元素也出栈。
5、当调用get_min时,返回栈B的栈顶元素,即栈A的最小值。

该方法实现了出栈(pop)、入栈(push)、取最小元素(get_min)的时间复杂度都是O(1),但是空间复杂度为O(n)

三、代码实现

class MinStack:
    def __init__(self):
        self.main_stack=[]
        self.min_stack
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值