LeetCode笔记：155. Min Stack

问题：

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.

大意：

设计一个栈，支持push、pop、top以及在固定时间内检索最小元素。

• push(x) -- 将元素x放入栈中。
• pop() -- 从栈的顶端移除元素。
• top() -- 获取栈顶元素。
• getMin() -- 检索栈中的最小元素 例子:

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.

思路：

这道题对于栈的操作本身都不难，并不是要用队列或者别的数据结构来模拟一个栈，直接就允许用栈来做。真正难的在于检索最小元素，并且还要再固定时间内。

要在入栈以及出栈的同时不停地计算最小元素，首先栈本身是不支持的，如果用数组之类的来记录，那么每次有新元素进来都需要排个序，非常耗时。

因此，这种方法巧妙地在每次入栈、出栈时进行简单的加减操作，达到可以直接获取最小元素的结果。在入栈时，入的不是实际的元素值，而是与当前记录的最小值的差值，如果新入的更小，就将其设为最小值，此时就保证了随时可以获取最小值。出栈时，要修改最小值。获取栈顶元素时，因为栈中记录的并不是原始值，所以要与记录的最小值进行操作来还原。

由于题目在submit时会用超过int范围的大数来测试，所以只能用long来操作。

他山之石（Java）：

public class MinStack {
    Stack<Long> stack;
    long min;

    /** initialize your data structure here. */
    public MinStack() {
        stack=new Stack<Long>();
    }
    
    public void push(int x) {
        if (stack.isEmpty()) {
            stack.push(0L);
            min = x;
        } else {
            stack.push(x - min);
            if (x < min) min = x;
        }
    }
    
    public void pop() {
        if (stack.isEmpty()) return;
        
        long pop = stack.pop();
        if (pop < 0) min = min - pop;
    }
    
    public int top() {
        long top = stack.peek();
        if (top < 0) return (int)min;
        else return (int)(min + top);
    }
    
    public int getMin() {
        return (int)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();
 */

合集：https://github.com/Cloudox/LeetCode-Record

查看作者首页