剑指 Offer 30. 包含min函数的栈（辅助栈&&对象的相等判断）

剑指 Offer 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.

提示：

1. 各函数的调用总次数不超过 20000 次

注意：本题与主站 155 题相同：https://leetcode-cn.com/problems/min-stack/

分析

要求获取最小元素的方法时间复杂度是O(1)，可以用一个辅助栈来辅助存储当前最小值。

如果进栈元素不大于当前的最小值，那么两个栈一起进栈。

如果进栈元素大于当前最小值，那么最小值不变，所以只需让这个元素进主栈，不进辅助栈。

弹出元素时:

如果主栈和辅助栈的栈顶不一致，那么主栈栈顶的元素一定大于辅助栈的，只需要弹出主栈栈顶即可。

如果主栈栈顶等于辅助栈栈顶，就把两个一起弹出。

特别注意：

1. 栈为空时，两个栈均进栈。题目不考虑栈空时的pop、top、min操作。
2. 对两个栈栈顶进行比较时，不能用等号，因为弹出的是Integer对象，只能用equals方法来比较是否相等。

代码

class MinStack {
    Stack<Integer> stack = null;
    Stack<Integer> helpStack = null;
    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<>();
        helpStack = new Stack<>();
    }
    
    public void push(int x) {
        //如果栈为空，就添加进去。
        if(stack.isEmpty()){
            stack.push(x);
            helpStack.push(x);
        }else{
            //栈不为空，如果x大于辅助栈，就只放进栈内
            if(x>helpStack.peek()) stack.push(x);
            else{
                //x小于等于辅助栈栈顶，就可以放进两个栈
                stack.push(x);
                helpStack.push(x);
            }
        }
    }
    
    public void pop() {
        //System.out.println(stack+"   "+helpStack);
        //如果栈顶和辅助栈栈顶一样，就一起弹出。
        if(stack.peek().equals(helpStack.peek())) {
            //System.out.println("一起弹出");
            stack.pop();
            helpStack.pop();
        }else{
            //反之只弹出栈顶
            stack.pop();
        }
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return helpStack.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.min();
 */