剑指offer题解（python3）----JZ30 包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的 min 函数，输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。

此栈包含的方法有：

push(value):将value压入栈中；

pop():弹出栈顶元素；

top():获取栈顶元素；

min():获取栈中最小元素；

数据范围：操作数量满足 0≤n≤300  ，输入的元素满足 ∣val∣≤10000；
进阶：栈的各个操作的时间复杂度是 O(1)，空间复杂度是 O(n)；

解析

首先要明确究竟如何从已经输入的栈中找到最小值。最直接的方法一定是想将所有元素压入栈后，对栈中的所有元素做遍历，但是题目要求所有操作的时间复杂度为O(1)，但没有任何一种遍历排序/查找操作的时间复杂度能降低到O(1)这样的水平。

因此就要想，寻找最小值最直接的需求就是比较，既然不能在同一个栈中进行比较，那么我们就在两个栈中进行互相比较。下列过程比较像栈计算器的比较，只不过栈计算器是比较运算符的优先级，但本题要求的是比较数。

栈2为空时，遇到一个数，先考虑将它压入栈1，和栈1栈顶元素进行比较，若小于栈1的栈顶元素，则压入栈2。

当栈2不为空时，遇到一个数，先考虑将它压入栈1，因为并不知道数是否为最小值；若该数小于栈2的栈顶元素，则将该数压入栈2中，否则压入栈1。

min方法即返回栈2的栈顶值，弹出即返回栈1的栈顶值。

代码

栈2即最小值栈。

class Solution:
    def __init__(self):
        self.stack = []  #初始化原栈
        self.stack_min = []  #初始化最小值栈
    def push(self, node):
        self.stack.append(node) #往原栈中添加 元素
        if not self.stack_min:    #一开始也往最小值栈中添加元素
            self.stack_min.append(node)
        elif node < self.stack_min[-1]:  #再次添加元素的时候要同最小值栈中元素比较，小于最小值栈中索引-1的对应值，就把node加入；
            self.stack_min.append(node)
        else:
            self.stack_min.append(self.stack_min[-1])  #否则把索引-1的对应值添加到stack_min中
    def pop(self):
        self.stack.pop()                    #弹出栈顶元素则分别获取即可
        self.stack_min.pop()
    def top(self):
        return self.stack[-1]
    def min(self):
        return self.stack_min[-1]