最小栈

设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。

  • push(x) -- 将元素 x 推入栈中。
  • pop() -- 删除栈顶的元素。
  • top() -- 获取栈顶元素。
  • getMin() -- 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

这道题很简单,也是常规操作,但是注意时间,需要注意一点的就是在检索栈的最小值的时候做优化,这里我是将每次压入栈的元素做最小值比较,将最小的值存入另外一个栈中。然后在寻找最小值的时候将他输出来就行。

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.__list = []
        self.__min = []
        

    def push(self, x):
        """
        :type x: int
        :rtype: void
        """
        self.__list.append(x)
        if self.__min == []:
            self.__min.append(x)
        else:
            if x <= self.__min[-1]:
                self.__min.append(x)
        

    def pop(self):
        """
        :rtype: void
        """
        if self.__list != [] and self.__min != []:
            if self.__min[-1] == self.__list[-1]:
                self.__min.pop()
        self.__list.pop()
        
        

    def top(self):
        """
        :rtype: int
        """
        return self.__list[-1] 
        

    def getMin(self):
        """
        :rtype: int
        """
        return self.__min[-1] if self.__min != [] else 0

如果代码还能优化,希望大家指出,或者有更好新的思路希望咱们也能讨论!

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值