最小栈
题目:最小栈
设计一个支持 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()