LeetCode No.155 Min Stack | #Stack

Q:

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.
· push(x) -- Push element x onto stack.
· pop() -- Removes the element on top of the stack.
· top() -- Get the top element.
· getMin() -- Retrieve the minimum element in the stack.
Example:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> Returns -3.
minStack.pop();
minStack.top(); --> Returns 0.
minStack.getMin(); --> Returns -2.

A:

如果是最小值,那么push进来两次,主要目的是在pop掉最小值的时候,能够还剩一个作为“存根”进行比对,这是很巧妙地思路,“存”+比对后,再把第二个pop掉。
测试用例: 6,-2,7,4,-6。
陆续push进去之后,stack里从上至下是:-6,-6,4,7,-2,-2,6,6。

class MinStack {
    Stack stack = new Stack<>(); //declaration one stack here
    int min = Integer.MAX_VALUE;  //先声明min, no worry about stack.empty

    public void push(int x) {
        if (x <= min) {
            stack.push(min); 
            min = x;         //单独记录下最小值是什么
        } 
        stack.push(x);       //(看上面的解释)
    } 

    public void pop() {
        int top = stack.pop();  //是赋值语句,也完成了pop操作
        if (top == min){
            min = stack.pop(); //之前每次出现min就push进两次,这里pop第二次
        }
    } //(看下面的代码)

    public int top() {
        return stack.peek();  //import java.util.Stack
    }

    public int getMin() {
        return min;
    }
}

下面这个实现 pop()的版本,含义完全一样,虽然不够简洁,但是更易懂
(备注里写了如何改动成上面精简版本的步骤和逻辑过程)

public void pop() { if (stack.peek() == min) { //2.要pop()肯定是针对stack.peek(),那直接stack.peek() = stack.pop(),然后判断是不是等于min
stack.pop(); //3. 第一步已经pop掉一次了,这个可以删了
min = stack.peek(); //4. 当初存两个目的就是为了这里,可以retrieve min value
stack.pop(); //5.已经知道min是啥了,可以把old min里的第二个“存根”也pop掉了
}
else { stack.pop();//1. 上面俩pop(),这里又一个pop(),说明无论如何都得pop()一次
} }

你可能感兴趣的:(LeetCode No.155 Min Stack | #Stack)