https://leetcode-cn.com/problems/dui-lie-de-zui-da-zhi-lcof/
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1.
class MaxQueue {
public:
MaxQueue() {}
int max_value() {}
void push_back(int value) {}
int pop_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();
*/
思路1:
增加成员变量,直接用std::queue,于是push_back和pop_front函数很容易实现,对于max_val函数,需要遍历整个队列得到最大值(时间复杂度为o(n),不太满足哦),另外std::queue不支持遍历,可以用一些技巧实现,
终止点的索引。
思路2:
思想:用一个辅助空间保存最大值,最大值的更新结合使用队列的特点。
增加两个成员变一个是单端队列std::queue<int>,用于存储放入的值;
一个是双端队列std::dque<int>用于维护最大值信息,该双端队列的头保持当前所有元素的最大值,后续的值们维护的信息是 当对头的最大值被pop之后,剩余元素的最大值。
如上,用户按照1 10 3 0 的顺序塞数字,当第一次输入1时,双端队列保存这个1,然后输入10, 由于10是最大的, 于是乎双端队列中把刚才的1剔除,保存10;
后续,用户输入3, 这个值需要放到双端队列的10的后面,0也是,因为,用户pop掉10之后,最大的就是3,pop掉3之后最大的就是0。
另外,如果用户紧接着输入数字7, 需要将双端队列中的队尾的所有比7小的都抹杀掉:
因为用户pop掉10之后,最大的就是7了。
如果用积木高度理解的话:如下图,队列的这个状态时,双端队列中只需要包保存99,30,20,10,其中99大哥用于笼罩其前方(A区)所有比他小点的值...
Code:
#pragma once
#include <iostream>
#include <queue>
using namespace std;
//https://leetcode-cn.com/problems/dui-lie-de-zui-da-zhi-lcof/
class MaxQueue {
public:
MaxQueue() {
}
int max_value() {
if (dqueCahcheMax.empty())
return -1;
else
return dqueCahcheMax.front();
}
void push_back(int value) {
queCacheAll.push(value); // 单端队列queue没有push_back,push_front函数,只能放到队尾
if (dqueCahcheMax.empty()) {
dqueCahcheMax.push_back(value);
}
else {
while ((!dqueCahcheMax.empty()) && (value > dqueCahcheMax.back())) {
dqueCahcheMax.pop_back();
}
dqueCahcheMax.push_back(value);//dqueCahcheMax的头始终保存当前所有数字的最大值
}
}
int pop_front() {
if (queCacheAll.empty())
return -1;
int res = queCacheAll.front();
queCacheAll.pop();
if (res == dqueCahcheMax.front()) {
dqueCahcheMax.pop_front();
}
return res;
}
private:
queue<int> queCacheAll;
deque<int> dqueCahcheMax;
};
void main() {
MaxQueue obj;
int param_3 = obj.pop_front();
obj.push_back(1);
obj.push_back(2);
int param_1 = obj.max_value();
return;
}