面试之手撕代码二

（二）
设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
解法：关键是如何在常数时间做到，解法为双栈（空间换时间），一个存数据，一个存最小值，push的同时，也在第二个栈push最小值（或者push的时候只比较当前值与最小栈的栈顶，不更大push,pop的同时判断是否与第二个栈顶相同，同则pop）。
Python版（第一种解法）：

class MinStack(object):

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

def push(self, x):
"""
:type x: int
:rtype: None
"""
self.result.append(x)
if len(self.min) == 0:
self.min.append(x)
else:
if x < self.min[-1]:
self.min.append(x)
else:
self.min.append(self.min[-1])
def pop(self):
"""
:rtype: None
"""
self.min.pop()
return self.result.pop()
def top(self):
"""
:rtype: int
"""
return self.result[-1]

def getMin(self):
"""
:rtype: int
"""
if len(self.min) == 0:
return []
return self.min[-1]

• 点赞
• 收藏
• 分享
• 文章举报