题目:
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
解答:
思路:
- 创建一个queue来存放数据,再创建一个deque来存放最大值。
- 每次添加新值时,queue正常offer进来,而deque要将deque里比新值小的值poll出去。从而保证deque是一个单调递减的队列。
- 每次删除旧值时,queue正常poll出去,而deque要判断队列的头部(即最大值)是否等于要删除的值,是的话就poll出去。不是的话说明在之前offer值的时候,当前要删除的值已经被删除了,所以现在就不用再删除了。
代码:
class MaxQueue {
Queue<Integer> queue;
Deque<Integer> deque;
public MaxQueue() {
queue = new LinkedList<>();
deque = new LinkedList<>();
}
public int max_value() {
return deque.isEmpty() ? -1 : deque.peekFirst();
}
public void push_back(int value) {
queue.offer(value);
while(!deque.isEmpty() && deque.peekLast() < value)
deque.pollLast();
deque.offerLast(value);
}
public int pop_front() {
if(queue.isEmpty()) return -1;
if(queue.peek().equals(deque.peekFirst()))
deque.pollFirst();
return queue.poll();
}
}
/**
* Your MaxQueue object will be instantiated and called as such:
* MaxQueue obj = new MaxQueue();
* int param_1 = obj.max_value();
* obj.push_back(value);
* int param_3 = obj.pop_front();
*/