题目描述:
请定义一个队列并实现函数 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;}
};
注意:上述代码并非原创,如原作者看到,可联系删除。