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

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

import java.util.Stack;

public class Solution {
/*思路：
利用一个辅助栈来存放最小值
    栈  3，4，2，5，1
    辅助栈 3，3，2，2，1
每入栈一次，就与辅助栈顶比较大小，如果小就入栈，如果大就入栈当前的辅助栈顶
当出栈时，辅助栈也要出栈
这种做法可以保证辅助栈顶一定都当前栈的最小值
*/
    Stack<Integer> data = new Stack<Integer>();
    Stack<Integer> min = new Stack<Integer>();

    public void push(int node) {
        data.push(node);
        if(min.isEmpty())
            min.push(node);
        else{
            if(node<=min.peek())
                min.push(node);
        }
    }

    public void pop() {
        if(data.peek()==min.peek())
            min.pop();
        data.pop();
    }

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

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

    public static void main(String[] args) {
        Solution s=new Solution();

        s.push(2);
        s.push(1);
        s.push(3);
        s.push(4);

        System.out.println(s.min());
        System.out.println(s.top());
    }
}