面试题30-包含min函数的栈

题目：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.min();   --> 返回 -2.

题目分析：

    栈是一个受限制的线性表，只能在栈顶端进行添加和删除。出栈和入栈时间复杂度可以达到 $O(1)$，但是要获取到栈中的最值，是要 $O(n)$ 的时间复杂度。这时候我们要考虑利用算法中常用的策略：用空间换时间。

    思路：我们借助一个辅助栈，存放与数据栈对应的最小值。数据栈存入17时，数据栈的最小值就是17，因此17进入辅助栈；后面元素同理；当数据栈存入9时，数据栈的最小值仍然是5，因此继续将5重复入栈一次，进入辅助栈。

Java代码：

import java.util.Stack;
/**
 * https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/
 *
 * 面试题30. 包含min函数的栈
 * 定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。
 *
 * 示例:
 *      MinStack minStack = new MinStack();
 *      minStack.push(-2);
 *      minStack.push(0);
 *      minStack.push(-3);
 *      minStack.min();   --> 返回 -3.
 *      minStack.pop();
 *      minStack.top();      --> 返回 0.
 *      minStack.min();   --> 返回 -2.
 */
public class Offer30 {

    private Stack<Integer> dataStack;    // 数据栈
    private Stack<Integer> minStack; // 辅助栈

    //　构造函数
    public Offer30(Stack<Integer> dataStack, Stack<Integer> minStack) {
        this.dataStack = dataStack;
        this.minStack = minStack;
    }

    // 入栈函数
    public  void push(int x) {
        // 无条件进入数据栈
        dataStack.push(x);
        // 如果当前辅助栈为空，我们无条件加入，
        // 如果非空，永远添加数据栈中的最小值

        // 当前x小于辅助栈的栈顶元素，说明x更小，需要进入辅助栈。
        if (minStack.isEmpty() || x<=minStack.peek())
            minStack.push(x);
        else
            // 如果不是更小，说明重复一下当前辅助栈栈顶元素
            minStack.push(minStack.peek());
        // 这样做的目的是为了出栈的时候可以都出栈一个元素。
    }

    public  void pop() {
        // 都出栈一个元素
        dataStack.pop();
        minStack.pop();
    }

    public  int top() {
        return dataStack.peek();
    }

    public  int min() {
        return minStack.peek();
    }

    public static void main(String[] args) {
        Stack<Integer> d = new Stack<Integer>();
        Stack<Integer> m = new Stack<Integer>();
        Offer30 minStack = new Offer30(d, m);
        minStack.push(-2);
        minStack.push(0);
        minStack.push(-3);
        System.out.println(minStack.min()); // -3
        minStack.pop();
        System.out.println(minStack.top());//   0
        System.out.println(minStack.min()); // -2
    }
}

【注】
（1）：leetcode 等平台只要我们完成一个函数即可，本人初出茅庐，为了巩固基本知识，故自己补充了部分代码，用于练手。本代码也许存在漏洞，望高手赐教。感谢！