请定义一个队列并实现函数 max_value 得到队列里的最大值
要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
难点:
对于普通队列,入队 push_back() 和出队 pop_front() 的时间复杂度均为 O(1) ;
本题难点为实现查找最大值 max_value() 的 O(1) 时间复杂度。
假设队列中存储 N 个元素,从中获取最大值需要遍历队列,时间复杂度为 O(N) ,单从算法上无优化空间。
考虑构建一个递减列表来保存队列 所有递减的元素 ,递减链表随着入队和出队操作实时更新,这样队列最大元素就始终对应递减列表的首元素,实现了获取最大值 O(1) 时间复杂度。
- 使用了
std::queue
作为存储数据的队列,使用了std::deque
(是双端队列,可以在队头和队尾进行插入和删除操作)作为辅助队列。 max_value()
函数用于返回当前队列中的最大值。如果辅助队列helpQueue
为空,说明队列中没有元素,返回-1;否则,返回helpQueue
的队头元素。push_back()
函数用于向队列中插入元素。我们首先将元素插入queue
队尾,然后从helpQueue
队尾开始,逐个比较并删除小于等于插入元素的元素,直到遇到大于插入元素的元素,然后将插入元素插入helpQueue
队尾。这样,helpQueue
中的元素始终保持递减的顺序。pop_front()
函数用于删除队列的队头元素。如果队列queue
为空,说明队列中没有元素,返回-1;否则,将queue
的队头元素保存为front
,然后将queue
的队头元素删除。如果queue
的队头元素front
等于helpQueue
的队头元素,说明队头元素是当前最大值,需要将helpQueue
的队头元素删除。最后,返回front
。
/**
* 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();
*/
class MaxQueue {
public:
MaxQueue() {
}
int max_value() {
if (helpQueue.empty()) {
return -1;
}
return helpQueue.front();
}
void push_back(int value) {
queue.push(value);
while (!helpQueue.empty() && helpQueue.back() < value) {
helpQueue.pop_back();
}
helpQueue.push_back(value);
}
int pop_front() {
if (queue.empty()) {
return -1;
}
int front = queue.front();
queue.pop();
if (front == helpQueue.front()) {
helpQueue.pop_front();
}
return front;
}
private:
queue<int> queue;
deque<int> helpQueue;
};