面试题 03.02. 栈的最小值（简单）

题目描述

请设计一个栈，除了常规栈支持的pop与push函数以外，还支持min函数，该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。

示例

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

做题思路

1.定义变量min来记录更新最小值；

2.定义valStack栈来进行所有值的入栈出栈操作；

3.定义minStack栈来进行存储最小值；

4.在返回最小值时，由于我们只需要返回最小值而无需关心中间值，所以在执行入栈操作时minStack栈只进行当前最小值与压入值中较小值的入栈，由于要保持两个栈中元素个数一致，在valStack出栈时minStack也要出栈。

代码

class MinStack {
    int min=Integer.MAX_VALUE; //记录最小值
    Stack<Integer> valStack=new Stack<>(); //维护所有值
    Stack<Integer> minStack=new Stack<>(); //维护最小值

    /** initialize your data structure here. */
    public MinStack() {

    }
    
    public void push(int x) {
        min=Math.min(min,x); //取最小值
        valStack.push(x); 
        minStack.push(min); //将最小值压入最小值栈中
    }
    
    public void pop() {
        valStack.pop();
        minStack.pop(); //元素出栈时最小值栈也要出栈，保证元素的个数一致
        if(minStack.isEmpty()){
            min=Integer.MAX_VALUE;
        }else{
            min=minStack.peek();
        }
    }
    
    public int top() {
        return valStack.peek();
    }
    
    public int getMin() {
        return min;
    }
}

/**
 * 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.getMin();
 */