难度中等
请定义一个队列并实现函数 max_value
得到队列里的最大值,要求函数max_value
、push_back
和 pop_front
的均摊时间复杂度都是O(1)。
若队列为空,pop_front
和 max_value
需要返回 -1
示例 1:
输入:
["MaxQueue","push_back","push_back","max_value","pop_front","max_value"]
[[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]
示例 2:
输入:
["MaxQueue","pop_front","max_value"]
[[],[],[]]
输出: [null,-1,-1]
限制:
1 <= push_back,pop_front,max_value的总操作数 <= 10000
1 <= value <= 10^5
使用队列实现,Java中Queue和Deque,队列和双端队列。
Queue接口作为队列数据结构,java在实现的时候,直接定义了Deque接口(双端队列)来继承Queue接口,并且只实现Deque接口。这样java中的双端队列就囊括了队列、双端队列、堆栈(Deque接口又定义了Stack的操作方法)这3种角色的功能。
Deque接口所能代表的数据结构:队列,双端队列,堆栈。
public class MaxQueue {
Queue<Integer> queue; // 增删数据
Deque<Integer> maxQueue;// 存最大值
public MaxQueue() {
queue = new ArrayDeque<>();
maxQueue = new ArrayDeque<>();// 双向队列
}
public int max_value() {
if (maxQueue.isEmpty()) return -1;
return maxQueue.peek();
}
public void push_back(int value) {
queue.add(value);
// 操作的是队列,添加队列时已经把之前队列最大值找出来,在队首。
// 后面的如果比max大,就替换掉。否则就放在max后面,这样即使把前面所有的都删除,最大值还会存在=新增的这个
// 注意:要求的只是均摊复杂度为O(1),理论上最快O(1),最慢O(n)
while(!maxQueue.isEmpty() && maxQueue.peekLast() < value){
maxQueue.pollLast();
}
maxQueue.add(value);
}
public int pop_front() {
if (queue.isEmpty()) return -1;
Integer poll = queue.poll();
if (poll.equals(maxQueue.peek())) {
maxQueue.poll();
}
return poll;
}
}