[菜鸟训练]155. 最小栈

题目描述：

设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。

示例:
输入：
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]

输出：
[null,null,null,null,-3,null,0,-2]

解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

提示：
pop、top 和 getMin 操作总是在 非空栈 上调用。

解题思路：

使用一个辅助栈，用来存取当前与正常栈中相对应得栈中最小值。我们每当入栈一个元素时，都去更新一下这个最小值，始终让它保持为当前最小的。所以，当我们正常的栈出栈一个元素的时候，我们的辅助栈也需要出栈一个元素。详见代码。

代码：

class MinStack {
    //辅助栈中的元素，一直用来记录与当前元素对应的最小值
    Stack<Integer> stack;
    Stack<Integer> minstack;
    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<>();
        minstack = new Stack<>();
    }
    
    public void push(int val) {
        if(stack.isEmpty()){
            stack.push(val);
            minstack.push(val);
        }
        else{
            stack.push(val);
            minstack.push(Math.min(val, minstack.peek()));
        }
    }
    
    //当一个元素要出栈的时候，我们将辅助栈中的栈顶元素一起弹出
    public void pop() {
        stack.pop();
        minstack.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int getMin() {
        return minstack.peek();
    }
}