剑指offer59 -II. 队列的最大值
一.题目
请定义一个队列并实现函数 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]
题目理解
二.解题思路
对于普通队列,入队 push_back() 和出队 pop_front() 的时间复杂度均为 O(1) ;本题难点为实现查找最大值 max_value() 的 O(1) 时间复杂度。
假设队列中存储 N 个元素,从中获取最大值需要遍历队列,时间复杂度为 O(N) ,单从算法上无优化空间。
如下图所示,最直观的想法是 维护一个最大值变量 ,在元素入队时更新此变量即可;但当最大值出队后,并无法确定下一个 次最大值 ,因此不可行。
考虑利用 数据结构 来实现,即经常使用的 “空间换时间” 。如下图所示,考虑构建一个递减列表来保存队列所有递减的元素 ,递减链表随着入队和出队操作实时更新,这样队列最大元素就始终对应递减列表的首元素,实现了获取最大值 O(1) 时间复杂度。
为了实现此递减列表,需要使用双向队列 ,假设队列已经有若干元素:
当执行入队 push_back() 时: 若入队一个比队列某些元素更大的数字 x ,则为了保持此列表递减,需要将双向队列尾部所有小于 x 的元素 弹出。
当执行出队 pop_front() 时: 若出队的元素是最大元素,则 双向队列需要同时将首元素出队 ,以保持队列和双向队列的元素一致性。
使用双向队列原因:维护递减列表需要元素队首弹出、队尾插入、队尾弹出操作皆为 O(1)O(1) 时间复杂度。
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); //类似于add(),若不存在,则返回false
while(!deque.isEmpty()&&deque.peakLast()<value)
deque.pollLast();//和remove()方法类似,返回flase,不抛异常,比较友好
deque.offerLast(value);
}
public int pop_front(){
if(queue.isEmpty()) return -1;
if (queue.peek().equals(deque.peekFirst()))/*在基本类型中,“==”比较的是两者的值;在引用类型中,“==”比较的是两者的内存地址,及是否为同一个对象;
equals比较两者的内存地址,及是否同一个对象,但在String类中自动重写,比较是两者的值是否相等;
如果此处用“==”,比较的是内存地址,用equals比较的是两者的值,因此用equals*/
deque.pollFirst();
return queue.poll();
}
}