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 操作总是在 非空栈 上调用。

解题思路

• 借用一个辅助栈min_stack，用于存获取stack中最小值。

• 算法流程：

  • push()方法： 每当push()新值进来时，如果 小于等于 min_stack栈顶值，则一起push()到min_stack，即更新了栈顶最小值；
  • pop()方法： 判断将pop()出去的元素值是否是min_stack栈顶元素值（即最小值），如果是则将min_stack栈顶元素一起pop()，这样可以保证min_stack栈顶元素始终是stack中的最小值。
  • getMin()方法： 返回min_stack栈顶即可。

• min_stack作用分析：

  • min_stack等价于遍历stack所有元素，把升序的数字都删除掉，留下一个从栈底到栈顶降序的栈。
  • 相当于给stack中的降序元素做了标记，每当pop()这些降序元素，min_stack会将相应的栈顶元素pop()出去，保证其栈顶元素始终是stack中的最小元素。

class MinStack {

    Stack<Integer> st;
    Stack<Integer> stm;

    /** initialize your data structure here. */
    public MinStack() {
        st = new Stack<>();
        stm = new Stack<>();
    }
    
    public void push(int val) {
        st.push(val);
        if (stm.isEmpty() || val <= stm.peek()) {
            stm.push(val);
        }
    }
    
    public void pop() {
        if (st.pop().equals(stm.peek())) {
            stm.pop();
        }
    }
    
    public int top() {
        return st.peek();
    }
    
    public int getMin() {
        return stm.peek();
    }
}

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