[230524 剑指59] 队列的最大值
一 题目
请定义一个队列并实现函数 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
二 整体思路
根据题意,我们要提供均摊时间复杂度为 O(1) 的 API。C++ 中的队列 queue 和 deque 本身提供的 push 和 back 都是 O(1) 时间复杂度的。所以我们要实现的就是在 O(1) 的时间复杂度内得到最大值。
如果我们只用一个 max 变量来记录当前队列中的最大值,会存在一个问题:当最大值出队后,我们无法知道队列里的下一个最大值。
我们可以使用一个辅助的数据结构:单调队列,队列的队头元素,就是当前队列中的最大值。在每次入队时,如果 deque 队尾元素小于即将入队的元素 value,则将小于 value 的元素全部出队后,再将 value 入队;否则直接入队。
三 关键点/重点/难点
关键点在维护单调栈,维护方法如上。
四 代码分析
class MaxQueue {
private:
queue<int> que;
deque<int> maxQue;
public:
MaxQueue() {
}
int max_value() {
if(maxQue.empty()) {
return -1;
}
return maxQue.front();
}
void push_back(int value) {
que.push(value);
while(!maxQue.empty() && maxQue.back() < value) {
maxQue.pop_back();
}
maxQue.push_back(value);
}
int pop_front() {
if(que.empty()) {
return -1;
}
int ret = que.front();
que.pop();
if(maxQue.front() == ret) {
maxQue.pop_front();
}
return ret;
}
};
(五)一题多解
应该没有吧。
(六) 知识扩展
C++中的 queue 和 的 deque:
总是忘记 queue 和 deque 的区别,强化一下记忆。
queue 是单向的,从队尾加入元素、从队首弹出元素,使用的接口分别是 push 和 pop。可以使用 front 和 back 访问队首和队尾元素。
deque 是双端的,可以从队尾加入和弹出元素、从队首加入和弹出元素,对应接口分别是 push_back 和 pop_back、push_front 和 pop_front。可以使用 front 和 back 访问队首和队尾元素。
deque 有 clear 接口而 queue 不支持。