剑指offer -- 包含min函数的栈(java)

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

思路

使用辅助栈存储最小值。定义两个栈：数据栈和辅助栈。数据栈存储真正压入的值，辅助站存储每次压栈的最小值。

每次往数据栈压入数据A，同时也往辅助栈压入数B，B = min(A,辅助栈的栈顶元素)。并且每次弹出数据栈栈顶元素，相应的，辅助栈也弹出栈顶元素。那么每次弹出数据栈元素时，辅助栈的栈顶元素永远是现有数据栈的最小值。

白话来讲就是，每次压入就存当时的最小值，所有的最小值都放到辅助栈里记录起来。每次弹出就是找到当时的真相，真相只有一个，就是辅助栈栈顶元素😂

出栈：

出栈轨迹（上图看懂的可以略过此图）：

实现

import java.util.Stack;

public class Solution {

    //数据栈
    private Stack stack = new Stack();
    //辅助栈
    private Stack minStack = new Stack();

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] nums = {3, 4, 7, 2, 3};
        solution.push(3);
        solution.push(4);
        solution.push(7);

        System.out.println(solution.min());
        solution.pop();
        System.out.println(solution.min());
        solution.pop();
        System.out.println(solution.min());
    }

    /**
     * 插入数据
     * @param node
     */
    public void push(int node) {
        stack.push(node);
        minPush(node);
    }

    /**
     * 往辅助栈插入数据
     * @param node
     */
    public void minPush(int node){
        //这个数和辅助栈做比较
        int minNum = node;
        if (minStack.size() > 0) {
            int peek = (Integer)minStack.peek();
            minNum = node < peek ? node : peek;
        }
        minStack.push(minNum);
    }

    /**
     * 弹出数据栈
     */
    public void pop() {
        stack.pop();
        minStack.pop();
    }

    /**
     * 获取数据栈最顶数据
     * @return
     */
    public int top() {
        return (Integer)stack.peek();
    }

    /**
     * 获取栈的最小值
     * @return
     */
    public int min() {
        int peek = (Integer)minStack.peek();
        return peek;
    }
}