1. 题目描述
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
2. 思路
本题主要解决的是找出队列的最大值,所以可以通过维护一个辅助队列(通过双向队列实现),使队头到队尾为单调递减的,则辅助队列的队头就是所求的最大值。另外一个队列用于入队、出队操作。
那怎么维护这样一个辅助队列呢?
当我们进行入队时,需要从队尾弹出比插入值小的元素,这些元素对答案是没有影响的;
当我们进行出队操作时,我们需要判断出队的元素是否为辅助队列的队头元素,如果是,辅助队列也要弹出队头元素。
3. 代码
class MaxQueue {
Queue<Integer> queue;//队列,用于入队、出队操作
Deque<Integer> helper;//辅助队列,用于获取最大值
public MaxQueue() {
queue = new LinkedList<Integer>();
helper = new LinkedList<Integer>();
}
//维护一个辅助队列,队头到队尾是单调递减的,从队头取出最大值
public int max_value() {
if(queue.isEmpty())
return -1;
return helper.peekFirst();
}
public void push_back(int value) {
queue.offer(value);
//将队尾比插入值还小的值移出辅助队列
while(!helper.isEmpty() && value > helper.peekLast()){
helper.pollLast();
}
//辅助队列入队
helper.offerLast(value);
}
public int pop_front() {
if(queue.isEmpty())
return -1;
Integer ans = queue.poll();
//如果出队的元素是辅助队列的队头元素,辅助队列也要移出队头元素
if(ans.equals(helper.peekFirst()))
helper.pollFirst();
return ans;
}
}