题目描述:
请定义一个队列并实现函数 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]
主要思想:
利用两个队列(一个单端队列q,一个双端队列d)。q用来存储队列中的值,d用来存储队列的最大值。
对于max_value()函数,当队列非空时,返回d的队首元素,否则返回-1。
这道题的关键在于最大值得存储。此处利用双端队列的优势,每次向队列中插入数据时,与d的队尾元素相比较,如果比队尾元素大,那么将队尾元素弹出队列,继续循环,最终维持双端队列d中的元素单调递减。
对于队列的弹出操作,如果q的首元素并非当前元素的最大值,直接弹出队列的值就可以了,如果为最大值时,要同时对两个队列执行弹出操作。
class MaxQueue {
queue<int> q;
deque<int> d;
public:
MaxQueue() {
}
int max_value() {
if (d.empty())
return -1;
return d.front();
}
void push_back(int value) {
while (!d.empty() && d.back() < value) {
d.pop_back();
}
d.push_back(value);
q.push(value);
}
int pop_front() {
if (q.empty())
return -1;
int ans = q.front();
if (ans == d.front()) {
d.pop_front();
}
q.pop();
return ans;
}
};
注意:上述代码并非原创,如原作者看到,可联系删除。