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

题目描述

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

注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

解题思路

第一反应：每次压入一个新元素进栈时，将栈里的所有元素排序，让最小的元素位于栈顶，这样就能在O(1)时间得到最小元素。但是不能保证最后压入栈的元素能够最先出栈，这个数据结构已经不是栈了。

在栈里添加一个成员变量存放最小的元素。每次压入一个新元素进栈的时候，如果该元素比当前最小的元素还要小，则更新最小元素。一个问题，如果当前最小的元素被弹出栈了，如何得到下一个最小的元素呢？

当最小元素被弹出栈的时候，我们希望能够得到次小元素，因此在压入这个最小元素之前，要把次小元素保存起来。

把每次的最小元素都保存起来放到另外一个辅助栈里呢？

首先往空的数据栈里压入数组3，显然现在3是最小值，也把这个最小值压入辅助栈。接下来往数据栈里压入数字4，由于4大于之前的最小值，因此我们仍然往辅助栈里压入数字3。第三步继续往数据栈里压入数字2，由于2小于之前的最小值3，把最小值更新为2，并把2压入辅助栈。同样当压入数字1时，也要更新最小值，并把新的最小值1压入辅助栈。

如果每次都把最小元素压入辅助栈，那么就能保证辅助栈的栈顶一直都是最小元素。当最小元素从数据栈内被弹出之后，同时弹出辅助栈的栈顶元素，辅助栈的新栈顶元素就是下一个最小值。

三个关键函数push，pop和min。m_data是数据栈，m_min是辅助栈

Python代码

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.min_stack = []
    def push(self, node):
        # write code here
        self.stack.append(node)
        if not self.min_stack or node <= self.min_stack[-1]:
            self.min_stack.append(node)
    def pop(self):
        # write code here
        if self.stack[-1] == self.min_stack[-1]:
            self.min_stack.pop()
        self.stack.pop()
    def top(self):
        # write code here
        return self.stack[-1]
    def min(self):
        # write code here
        return self.min_stack[-1]