【剑指Offer-59-II】队列的最大值
问题
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
示例
输入:
["MaxQueue","push_back","push_back","max_value","pop_front","max_value"]
[[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]
解答
class MaxQueue {
public:
int max_value() {
return q.empty() ? -1 : d.front();
}
void push_back(int value) {
q.push(value);
while (!d.empty() && value > d.back()) d.pop_back();
d.push_back(value);
}
int pop_front() {
if (q.empty()) return -1;
int res = q.front(); q.pop();
if (res == d.front()) d.pop_front();
return res;
}
private:
deque<int> d;
queue<int> q;
};