【LeetCode】 155. 最小栈

题目

题目传送门：传送门（点击此处）

题解

思路

在题解看了一圈也没有找到java中用 Stack 和 PriorityQueue 实现的，所以就把自己的代码贴上来仅供参考吧

push(x)： Stack push 时间复杂度 O(1)，PriorityQueue push 时间复杂度 O(lgn)
pop()： Stack pop 时间复杂度 O(1)，PriorityQueue pop 时间复杂度 O(lgn)
top()： Stack top 时间复杂度 O(1)
getMin()： PriorityQueue getMin 时间复杂度 O(lgn)

code

class MinStack {

    PriorityQueue<Integer> minHeap;
    Stack<Integer> stack;

    /**
     * initialize your data structure here.
     */
    public MinStack() {
        minHeap = new PriorityQueue();
        stack = new Stack<>();
    }

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

    public void pop() {
        int x = stack.pop();
        minHeap.remove(x);
    }

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

    public int getMin() {
        return minHeap.peek();
    }
}