【栈操作】Leetcode 面试题 03.02. 栈的最小值

【栈操作】Leetcode 面试题 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.

思路

• 设计两个栈
• 一个栈用来保存当前栈中的元素，其功能和一个正常的栈没有区别，这个栈记为stackData,另一个栈用来保存每一步的最小值，这个栈记为stackMin
• 每一次入栈的元素都和stackMin的栈顶元素进行比较，如果小于等于栈顶元素，直接入栈，每一次stackData出栈元素和statckMin栈顶元素进行比较，如果相等，stackMin出栈顶元素

代码

class MinStack {

    private Stack<Integer> stackData;
    private Stack<Integer> stackMin;

    /** initialize your data structure here. */
    public MinStack() {
        // 初始化 私有成员
        this.stackData = new Stack<>();
        this.stackMin = new Stack<>();
        this.stackMin.push(Integer.MAX_VALUE);// 入栈一个最大值
    }
    
    // 入栈操作
    public void push(int x) {
        // 首先StackData 肯定要入栈
        // 对于当前需要入栈的元素 与stackMin栈顶元素进行比较 如果小于或者等于 直接入栈
        // 如果satckMin栈空 直接入栈
        if(this.stackMin.isEmpty())
        {
            // 如果栈空  直接 入栈
            this.stackMin.push(x);
        }
        if(this.stackMin.peek() >= x)
        {
            this.stackMin.push(x);// 小于或者等于 直接入栈
        }
        this.stackData.push(x);
    }
    
    public void pop() {
        // 如果stackMin栈空 直接抛出异常

    
        int value = this.stackData.pop();

        // 同时更新
        if(value == this.stackMin.peek())
        {
            //如果弹出的元素等于satckMin()栈顶元素 说明弹出的是最小值 更新最小值栈
            this.stackMin.pop();// 直接pop最小值 更新最小值
        }
    }
    
    public int top() {
        // 获取栈顶元素
        return this.stackData.peek();
    }
    
    public int getMin() {
        // // 先判断stackMin是不是空栈
        // if(this.stackMin.isEmpty())
        // {
        //     // throw new RunTimeException("Your stack is empty");
        // }
        return this.stackMin.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.getMin();
 */