剑指Offer之队列的最大值

题目描述:

请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。若队列为空,pop_front 和 max_value 需要返回 -1。


思路分析:

该题和实现最小栈类似。我们可以设置一个辅助队列

1)当元素进队列时,元素进队列的同时,判断:如果当前进队元素value > 辅助队列中的某些元素,辅助队列尾部小于value的元素循环出队,最后较大的value进入辅助队列。这一步完成后辅助队列里存放的元素由大到小排序。

2)当元素出队列时,元素出队列的同时,判断出队元素值和辅助队列的值是否相等,如果相等,则辅助队列的队尾元素也要出队。

3)求最大值时,只需返回辅助队列中的队头元素即可。

参考代码:

 
class MaxQueue {
    private LinkedList<Integer> queue;
    private LinkedList<Integer>  maxQueue;
   
    public MaxQueue() {
         this.queue = new LinkedList<>();
         this.maxQueue = new LinkedList<>();
    } 
 
    public void push_back(int value) {
        queue.addLast(value);
        while(!maxQueue.isEmpty() && value > maxQueue.peekLast()){
                maxQueue.removeLast();
         }
         maxQueue.addLast(value);
    }

    public int pop_front() {
       if(queue.isEmpty())return -1;
       if(!maxQueue.isEmpty() && queue.peek().equals(maxQueue.peek()))maxQueue.removeFirst();
       return queue.removeFirst();
    }

    public int max_value() {
       if(maxQueue.isEmpty())return -1;
       return maxQueue.peek();
    }
} 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章