题目
请定义一个队列并实现函数 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
解题
主要思路:使用单独的队列维持单调递减的队列,保证max_value()方法获取的最大值。
class MaxQueue {
private Deque<Integer> deque;
private Deque<Integer> deque1;
public MaxQueue() {
deque = new LinkedList<Integer>();
deque1 = new LinkedList<Integer>();
}
// 返回队列中的最大值 为空返回-1
// 最大值必须要在队列里面,如果最大值被pop,需要更新最大值
public int max_value() {
return deque1.isEmpty() ? -1 :deque1.peekFirst();
}
// 往队尾插入元素
public void push_back(int value) {
deque.add(value);
while(!deque1.isEmpty() && deque1.peekLast() < value) {
deque1.removeLast();
}
deque1.add(value);
}
// 从队首获取元素 为空返回-1
public int pop_front() {
int res = deque.isEmpty() ? -1 : deque.pollFirst();
if (!deque1.isEmpty() && res == deque1.peekFirst()) deque1.removeFirst();
return res;
}
}