题目描述:
请定义一个队列并实现函数 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();
}
}