面试题21. 包含min函数的栈

题目描述
定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路：
考察栈的使用。代码中用到两个栈：
a. 用栈 stack 记录所有元素的值
b. 用辅助栈 min_stack 始终记录每次压栈时的最小元素，min_stack的栈顶就是当前的最小元素

具体操作如下：

• 假设我们要把数据A入栈，
  
  • 如果A小于等于 min_stack 的栈顶元素，则同时向stack和min_stack压入A
  • 否则，向min_stack压入min_stack的栈顶元素，向stack压入A
• 出栈时，stack和min_stack同时出栈

java 代码如下：

import java.util.Stack;public class Solution {    Stack<Integer> stack = new Stack();    Stack<Integer> min_stack = new Stack();    public void push(int node) {        if(min_stack.isEmpty() || node <= min_stack.peek()) {            min_stack.push(node);        }else {            min_stack.push(min_stack.peek());        }        stack.push(node);    }    public void pop() {        min_stack.pop();        stack.pop();    }    public int top() {        return stack.peek();    }    public int min() {        return min_stack.peek();    }}

python 代码如下：

# -*- coding:utf-8 -*-class Solution:    stack = []    min_stack = []    def push(self, node):                if len(self.stack) == 0 or node <= self.stack[-1]:            self.min_stack.append(node)        else:            self.min_stack.append(self.min())        self.stack.append(node)    def pop(self):        self.min_stack.pop()        self.stack.pop()    def top(self):        return self.stack[-1]    def min(self):        return self.min_stack[-1]