题目描述
请定义一个队列并实现函数 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
方法一(双端队列)
1.解题思路
- 定义两个双端队列,一个用于存取数据,一个用于快速获取最大值。
- 入队时,data队列直接入队,max队列按从大到小的顺序入队,如果不满足,直接从队尾出队,保证队头的元素最大。
- 出队时,首先判断data队列是否为空,如果为空,直接返回-1;然后判断data队头是否是最大值,如果是,则max队列要弹出队头。
- 获取最大值,先判断data是否为空,不为空则直接返回max队头。
2.代码实现
class MaxQueue {
Deque<Integer> data;
Deque<Integer> max;
public MaxQueue() {
data=new LinkedList<>();
max=new LinkedList<>();
}
public int max_value() {
if(data.isEmpty()) return -1;
return max.getFirst();
}
public void push_back(int value) {
data.addLast(value);
while(!max.isEmpty()&&max.getLast()<value){
max.removeLast();
}
max.addLast(value);
}
public int pop_front() {
if(data.isEmpty()) return -1;
int value=data.removeFirst();
if(max.getFirst()==value){
max.removeFirst();
}
return value;
}
}
3.复杂度分析
- 时间复杂度:显然max_value()和pop_front()的时间复杂度为O(1),push_back()操作看起来有循环,但是做一次push操作,最多有n个元素入队,而出队的时间复杂度为O(1),也就是每个元素只会出队1次,所以均摊的时间复杂度为O(1)。
- 空间复杂度:需要额外O(n)队列来存储元素,所以空间复杂度为O(n)。
剑指offer全集入口: 请戳这里