包含min函数的栈

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

解题思路：定义两栈stack1和stack2，stack1用来入栈，stack2只负责输出栈里最小数。

	public void push(int x) {
       /**
       	直接入stack1，然后再判断stack2是否为空，或者stack2的栈顶元素
       	大于等于x，如果条件满足，则x也入stack2
       */
    }
    
    public void pop() {
        /**
       	出栈时直接判断stack1和stack2栈顶元素是否相等，相等的话一起出栈
       	负责只出stack1栈顶元素。
       */
    }
    
    public int top() {
        /**
       	直接入s输出stack1栈顶元素即可
       */
    }
    
    public int min() {
        /**
	     直接输出stack2栈顶元素即可
       */
    }

下面附上代码

class MinStack {
    Stack<Integer> stack1;
    Stack<Integer> stack2;
    public MinStack() {
		stack1 = new Stack();
		stack2 = new Stack();
    }
    
    public void push(int x) {
        stack1.push(x);
        if(stack2.empty() || stack2.peek() >= x){
            stack2.push(x);
        }
    }
    
    public void pop() {
        if(stack1.pop().equals(stack2.peek()))
            stack2.pop();
    }
    
    public int top() {
        return stack1.peek();
    }
    
    public int min() {
        return stack2.peek();
    }
}