【剑指Offer】20.Python实现包含min函数的栈

问题描述：

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

思路：

借助一个辅助栈minstack用来实时存储stack中的最小元素，来实现stack的min函数，即要定义两个栈。

1. 实现入栈push：

往stack里压入node之前，首先要判断当前栈的最小元素是否存在，如果不存在，要把node先压入minstack中，表示node就是node入栈后的最小元素。如果当前栈的最小元素存在，要比较新入栈元素node和当前栈的最小元素的大小关系，谁小，就把谁压入辅助栈minstack。

2. 实现出栈pop()

在元素 pop() 出栈之前，要先把辅助栈 minstack 中的元素 pop() 出一个来，因为 minstack 中的元素，是随着stack元素更新而更新的。stack入栈一个元素，minstack就会入栈一个当前stack的最小元素。

3.返回最后一个入栈元素top

这个函数不是元素出栈，而仅仅是返回 stack[-1] ，即可; 当 stack 为空时，返回None。

4.实现min函数，返回当前栈的最小值。

由于辅助栈 minstack 就是存储的栈的最小元素，因此当 minstack 非空时，直接返回 minstack[-1] 即可。

具体代码实现：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.minstack = []
    def push(self, node):
        # write code here
        min = self.min()     # 查看当前栈的最小元素
        if not min or node < min: # 如果当前栈没有最小值或者是入栈元素要小于当前栈的最小值，将node压入minstack
            self.minstack.append(node)
        else:    # 否则就表示新元素入栈之后，占得最小元素仍然是当前栈的最小值
            self.minstack.append(min)
        self.stack.append(node)  # 不管怎样都要将node压入栈中
    def pop(self):
        # write code here
        if self.stack:      # 检查栈是否非空
            self.minstack.pop()  # 两者同时更新
            return self.stack.pop()
        return None
    def top(self):
        # write code here
        if self.stack:
            return self.stack[-1]
        return None
    def min(self):
        # write code here
        if self.minstack:
            return self.minstack[-1]
        return None

代码测试：

# -*- coding:utf-8 -*-
'''
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。
'''
class Solution:
    def __init__(self):
        #定义两个栈，其中一个辅助栈保存最小值
        self.stack=[]
        self.minstack=[]
    def push(self, node):
        # write code here
        #查看当前栈的最小元素
        min=self.min()
        print("当前栈的最小值为："+str(min))
        #如果当前栈没有最小值或者是入栈元素要小于当前栈的最小值
        #将node压入minstack中
        if not min or node<min:
            self.minstack.append(node)
        else:
            #否则就表示新元素入栈之后，占得最小元素仍然是当前栈的最小值
            self.minstack.append(min)
        self.stack.append(node) #不管怎样将node压入栈中
        print("添加元素"+str(node)+"之后，当前栈为："+str(self.stack))
        print("添加元素"+str(node)+"之后，当前辅助栈（保存最小值）为：" + str(self.minstack))
    def pop(self):
        # write code here
        if self.stack:
            #两者同时更新
            self.minstack.pop()
            return self.stack.pop()
        return None
    def top(self):
        # write code here
        if self.stack:
            return self.stack[-1]
        return None
    def min(self):
        # write code here
        if self.minstack:
            return self.minstack[-1]
        return None

if __name__=='__main__':
    minstack=Solution()
    minstack.push(3)
    minstack.push(2)
    minstack.push(5)
    minstack.push(1)
    print('当前栈的最小值为：'+str(minstack.min()))
    #minstack.pop()
    print('删除栈顶元素'+str(minstack.pop())+'后，当前栈的最小值为：'+str(minstack.min()))
    #minstack.pop()
    print('删除栈顶元素' + str(minstack.pop()) + '后，当前栈的最小值为：' + str(minstack.min()))
    #minstack.pop()
    print('删除栈顶元素' + str(minstack.pop()) + '后，当前栈的最小值为：' + str(minstack.min()))

运行结果如下：