LeetCode155. 最小栈

又是昨天的，这两天有点儿懒，栈的实现，很简单的。

题目

设计一个支持 push，pop，top 操作，并能在常量时间内检索最小元素的栈。

• push(x) -- 将元素x推入栈中。
• pop() -- 删除栈顶的元素。
• top() -- 获取栈顶元素。
• getMin() -- 检索栈中的最小元素。

示例:

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

分析

像是入栈出栈获取栈顶元素，都没啥好说的，就直接用了java已经提供了的方法。

getMin（）检索栈中的最小元素，一开始想着用一个min临时int变量来存一下最小的那个数字，在调用getMin（）的时候，直接返回min，但是这样并没有考虑到出栈的问题，如果当前最小的数字出栈了，而min并没有办法及时更新，所以换成了优先队列，来保存最小元素的信息，当出栈的时候，将出栈的元素删掉，就可以了。

代码

class MinStack {

    Stack<Integer> stack = new Stack<>();
    PriorityQueue<Integer> pq = new PriorityQueue<>();

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

    }

    public void push(int x) {
        stack.add(x);
        pq.add(x);
    }

    public void pop() {
        int temp = stack.peek();
        pq.remove(temp);
        stack.pop();
    }

    public int top() {
        return stack.peek();
    }

    public int getMin() {
        return pq.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();
 */