请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。若 队列为空,pop_front 和 max_value 需要返回 -1
分析:
对于普通队列,入队 push_back() 和出队 pop_front() 的时间复杂度均为 O(1) ;本题难点为实现查找最大值 max_value() 的 O(1)时间复杂度。假设队列中存储 N 个元素,从中获取最大值需要遍历队列,时间复杂度为 O(N) ,单从算法上无优化空间。
考虑利用 数据结构 来实现,即经常使用的 “空间换时间” 。如下图所示,考虑构建一个递减列表来保存队列 所有递减的元素 ,递减链表随着入队和出队操作实时更新,这样队列最大元素就始终对应递减列表的首元素,实现了获取最大值 O(1)时间复杂度
使用队列queue存放数据,可以从队尾插入,队首弹出
使用双向队列deque放置递减的最大值来对应queue中的数据,可以使用尾部删除pop_back
(使用双向队列原因:维护递减列表需要元素队首弹出、队尾插入、队尾弹出操作皆为 O(1) 时间复杂度。)
不难理解,当queue弹出值的大小和deque的头部元素大小相同时,两者需要同时弹出,才不会影响后续最大值结果
函数设计:
初始化队列 queue ,双向队列 deque ;
入队 push_back() :
将元素 value 入队 queue ;
将双向队列中队尾 所有 小于 value 的元素弹出(以保持 deque 非单调递减),并将元素 value 入队 deque ;
出队 pop_front() :
若队列 queue 为空,则直接返回 -1−1 ;
否则,将 queue 首元素出队;
若 deque 首元素和 queue 首元素 相等 ,则将 deque 首元素出队以保持两队列 元素一致
最大值 max_value() :
当双向队列 deque 为空,则返回 -1−1 ;
否则,返回 deque 首元素;
class MaxQueue {
public:
queue<int> q;
deque<int> d_max;
int temp = 0;
MaxQueue() {}
int max_value() {
if(d_max.empty()){
return -1;
}
else{
return d_max.front();
}
}
void push_back(int value) {
q.push(value);
while(!d_max.empty()&&(d_max.back()<value)){
d_max.pop_back();
}
d_max.push_back(value);
}
int pop_front() {
if(q.empty()){
return -1;
}
else{
temp = q.front();
if(q.front()==d_max.front()){
d_max.pop_front();
}
q.pop();
return temp;
}
}
};
/**
* 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();
*/