请定义一个队列并实现函数 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]
示例 2:
输入:
[“MaxQueue”,“pop_front”,“max_value”]
[[],[],[]]
输出: [null,-1,-1]
限制:
1 <= push_back,pop_front,max_value的总操作数 <= 10000
1 <= value <= 10^5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/dui-lie-de-zui-da-zhi-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
辅助队列,和能够返回最小值的栈思路有些类似,但也不相同,对栈来说,新入栈的元素只会影响自己当前的最小或者最大值,因此只需要每次入栈的时候记录下当前的最小或者最大值即可。但是队列来说,后面入队列的元素会影响前面元素的最大最小值,因此,需要使用一个辅助队列,辅助队列中根据元素入队列的顺序维护一个单调递减的队列,每次获取队头元素即为当前最大值,当出队列的时候,主队列的元素和辅助队列的元素相等的时候,辅助队列队头元素才出队列。
逻辑是这样的,相当于辅助队列维护的是阶段最大值,模糊掉每个阶段的最值之间的值,那么就获取了每个阶段的最值。
class MaxQueue {
public:
vector<int> q1;
vector<int> q2;
int head1;
int head2;
int len;
int tail;
MaxQueue() {
head1 = 0;
head2 = 0;
len = 0;
tail = -1;
}
int max_value() {
cout<<"max"<<endl;
if(len==0)return -1;
return q2[head2];
}
void push_back(int value) {
cout<<"push"<<endl;
q1.push_back(value);
while(tail>=head2&&value>q2[tail])
{
q2.pop_back();
tail--;
}
q2.push_back(value);
tail++;
len++;
}
int pop_front() {
cout<<"pop"<<endl;
if(len==0)return -1;
int ret = q1[head1];
head1++;
len--;
if(ret==q2[head2])
{
head2++;
}
return ret;
}
};
/**
* 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();
*/