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();
 */

阅读更多
个人分类: 算法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

LeetCode155. 最小栈

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭