python算法:如何用O(1)的时间复杂度求栈中最小元素

class Stack:
    def __init__(self):
        self.items = []

    def empty(self):
        return len(self.items) == 0

    def size(self):
        return self.items

    def peek(self):
        if not self.empty():
            return self.items[-1]
        else:
            return None

    def pop(self):
        if len(self.items)>0:
            return self.items.pop()
        else:
            return None

    def push(self,item):
        self.items.append(item)

class MyStack:
    def __init__(self):
        #用来存储栈中元素
        self.elemStsck = Stack()
        #栈顶永远存储在当前elemStack中最小值
        self.minStack = Stack()

    def push(self,data):
        self.elemStsck.push(data)
        if self.minStack.empty():
            self.minStack.push(data)
        else:
            if data<self.minStack.peek():
                self.minStack.push(data)

    def pop(self):
        topData = self.elemStsck.peek()
        self.elemStsck.pop()
        if topData == self.mins():
            self.minStack.pop()
        return topData

    def mins(self):
        if self.minStack.empty():
            return None
        else:
            return self.minStack.peek()

if __name__ == '__main__':
    stack = MyStack()
    print(stack.mins())
    stack.push(3)
    print(stack.mins())

这种方法额外申请一个栈的空间来保存栈中最小的元素,从而达到O(1)时间复杂度求栈中最小元素目的,担付出空间的复杂度为O(N)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值