包含min函数的栈（Java）

包含min函数的栈（Java）

题目：
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））
分析：
设置两个栈，一个用来存放数据，一个用来存放当前的最小值。
每次push一个数时，如果为空，则两个栈都push。如果非空，data栈push，min栈比较与栈顶元素的大小，push较小的那个进栈。

import java.util.Stack;

/**
 * @desc 包含min函数的栈
 * @author zhaoliang
 * @date 20200506
 */
public class Main20 {
    //定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
    //注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。
    private Stack<Integer> dataStack = new Stack<>();
    private Stack<Integer> minStack = new Stack<>();
    public void push(int node) {
            dataStack.push(node);
            if (minStack.isEmpty() ||minStack.peek() >dataStack.peek()){
                minStack.push(dataStack.peek());
            }else {
                minStack.push(minStack.peek());
            }
    }

    public void pop() {
        if (!dataStack.isEmpty()){
            dataStack.pop();
        }
        if (!minStack.isEmpty()){
            minStack.pop();
        }
    }

    public int top() {
        //  取出数据栈的栈顶元素
        return dataStack.peek();
    }

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