题目
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
思考过程:
我们的题目中有一个要求:时间复杂度为O(1)。
队列push_back 和 pop_front的操作时间复杂度本来就是O(1),所以我们只需要是max_value()做到时间复杂度为O(1)。
第一时间想到的就是辅助队列(类似之前的辅助栈)。
但稍微一思考,若队列为654321,辅助栈中只有6,到队首元素出队时,辅助队列中的首元素也出列,我们就不知道接下来谁是最大的,所以不可行。
所以我们又想到,一个递减的辅助队列。
为了实现此递减列表,需要使用 双向队列 ,假设队列已经有若干元素:
当执行入队 push_back() 时: 若入队一个比队列某些元素更大的数字 x ,则为了保持此列表递减,需要将双向队列 尾部所有小于 x 的元素 弹出。
当执行出队 pop_front() 时: 若出队的元素是最大元素,则 双向队列 需要同时 将首元素出队 ,以保持队列和双向队列的元素一致性。
使用双向队列原因:维护递减列表需要元素队首弹出、队尾插入、队尾弹出操作皆为 O(1)时间复杂度。
设计双向队列为 非单调递减 的原因:若队列 queue 中存在两个 值相同的最大元素 ,此时 queue 和 deque 同时弹出一个最大元素,而 queue 中还有一个此最大元素;即采用单调递减将导致两队列中的元素不一致。
题解:
class MaxQueue {
queue<int> que;
deque<int> deq;
public:
MaxQueue() {
}
int max_value() {
if(deq.empty())
return -1;
return deq.front();
}
void push_back(int value) {
que.push(value);
while(!deq.empty() && deq.back() < value)
deq.pop_back();
deq.push_back(value);
}
int pop_front() {
if(que.empty()) return -1;
int val = que.front();
if(val == deq.front())
deq.pop_front();
que.pop();
return val;
}
};
/**
* 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();
*/