剑指offer第30题:包含min函数的栈
题目:定义栈的数据结构,实现一个能够获得最小值的min函数。在该栈中,调用min、push及pop的时间复杂度都是O(1)。
这题如果用python的min函数,解决十分轻松,但可能不符合它的时间复杂度,我们自己实现栈,并完成min函数。
# _*_encoding:utf-8 _*_
class Stack:
"""
用列表表示栈,符合先进后出的原则
用两个栈解决:
第一个栈stack正常出入栈
第二个support_stack每次push时,将stack中的最小值push进去,而不是直接push某一个值
每次pop操作,两个栈都进行pop,
这样support_stack[-1]表示当前stack的最小值
"""
def __init__(self):
self.stack = []
self.support_stack = []
# min_num其实可以不要,用support_stack[-1]表示即可
self.min_num = None
def push(self, num):
self.stack.append(num)
if self.min_num is None:
self.min_num = num
self.support_stack.append(self.min_num)
elif self.min_num < num:
self.support_stack.append(self.min_num)
else:
self.min_num = num
self.support_stack.append(num)
def pop(self):
if self.stack:
self.support_stack.pop()
return self.stack.pop()
return
def min(self):
return self.support_stack[-1]
if __name__ == '__main__':
stack = Stack()
current_stack = [11, 7, 8, 9, 5, 12, 15]
for num in (current_stack):
stack.push(num)
print stack.min(), stack.stack
# 测试用例
stack.push(7)
# 输入一个比小值大的
print stack.min(), stack.stack
stack.push(4)
# 输入一个比小值小的
print stack.min(), stack.stack
stack.pop()
# 弹出最小值
print stack.min(), stack.stack
stack.pop()
# 弹出非最小值
print stack.min(), stack.stack