30：包含min函数的栈（剑指offer第2版Python）

1、题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

 

2、代码详解

引入两个栈，一个栈（数据栈）每次push实际的数字，另一个minStack（辅助栈）。
如果push的数字小于minStack栈顶的数字，push新的数字，反之，把栈顶的数字再压入一遍。

代码如下：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = [] # 数据栈
        self.minStack = [] # 辅助栈
    def push(self, node):
        self.stack.append(node)
        # 如果node比当前最小的小，将“node”压栈到辅助栈
        if self.minStack == [] or node < self.min():
            self.minStack.append(node)
        # 如果node大于当前最小的，将“原最小值”压栈到辅助栈
        else:
            temp = self.min()
            self.minStack.append(temp)

    def pop(self):
        if self.stack == [] or self.minStack == []:
            return None
        self.minStack.pop()
        self.stack.pop()

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

    def min(self):
        return self.minStack[-1]



S = Solution()
S.push(3)
S.push(4)
S.push(2)
S.push(1)

print(S.min())
S.pop()
print(S.min())
S.pop()
print(S.min())