LeetCode155-最小栈(优先队列/巧妙的解法)

时间：2018-08-10 23:07:02 阅读：146 评论：0 收藏：0 [点我收藏+]

标签：图片 return [] queue image 是的自定义优先 style

看起来挺简单，但是写起来才有坑。

模仿java里面的栈

1、用数组存放元素

2、设置size和index，push和pop只需要移动index就好了，不需要处理元素。

3、初始化为16，如果满了要扩容到2倍，为了偷懒，数组只增不减。

最后就是处理min的问题，原来想着提供一个min变量，每次插入的时候更新min即可。

但是如果刚好pop了一个min呢?上一个min是多少?

当然可以使用一个secondMin变量，但是连续pop两个呢？上上个呢？

所以使用变量不行。

因为只需要在获取的时候是常数，维护min不做限制，那么使用优先队列就好了。

注意在pop的时候，要注意pop的是不是最小值，是的话有限队列也要poll

没进行出错处理，比如pop到负数，但是题目本身也没这样的操作。过了

class MinStack {

    int size;
        int index;

        //只有一个min，或者两个min，不能实现该功能
        //如果pop了一个min，没法获取之前的最小值了

        //使用最小堆就好了
        PriorityQueue<Integer> min;

        int [] array;


            /** initialize your data structure here. */
            // 初始化给16
            public MinStack() {
                size = 16;
                index = 0;
                array = new int [16];
                min = new PriorityQueue<>();
            }

            public void push(int x) {
                //扩容
                if(index==size-1){
                    size*=2;
                    int [] temp = new int [size];
                    int ori = size/2;
                    for(int i=0;i<ori;i++){
                        temp[i]=array[i];
                    }
                    array = temp;
                }

                array[index++]=x;
                min.add(x);

            }

            public void pop() {
                int temp = array[--index];
                //如果pop掉的是最小值，那么队列也要删掉一个最小值
                if(temp==getMin()){
                    min.poll();
                }
            }

            public int top() {
                return array[index-1];
            }

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