最小栈

题目:最小栈

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

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

解答:

1)最粗暴的解法,用list完成

2)用两个栈,一个栈记录全部元素,一个栈记录最小元素

代码

方案一:用list完成

class Solution(object):
    def __init__(self, stack):
        self.stack = stack
        
    def push(self, x):
        self.stack.append(x)

    def pop(self):
        self.stack.pop()

    def top(self):
        return self.stack[-1]

    def get_min(self):
        return min(self.stack)

方案二:两个栈

class Solution(object):
def __init__(self):
self.stack = []
self.min_stack = []

def push(self, x):
self.stack.append(x)
if self.min_stack:
if x <= self.min_stack_top():
self.min_stack.pop()
self.min_stack.append(x)
else:
self.min_stack.append(x)

print(self.stack)
print(self.min_stack)

def min_stack_pop(self):
self.min_stack.pop()

def min_stack_top(self):
return self.min_stack[-1]

def pop(self):
self.stack.pop()

def top(self):
return self.stack[-1]

def get_min(self):
return self.min_stack_top()

posted on 2018-12-24 11:13  罗盼  阅读(108)  评论(0编辑  收藏  举报

导航