题目
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
输入:
[“MaxQueue”,“push_back”,“push_back”,“max_value”,“pop_front”,“max_value”]
[[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]
思路
借助一个辅助队列,使其始终保持单调递减的特性。
本算法基于问题的一个重要性质:当一个元素进入队列的时候,它前面所有比它小的元素就不会再对答案产生影响。
如果我们向队列中插入数字序列 1 1 1 1 2,那么在第一个数字 2 被插入后,数字 2 前面的所有数字 1 将不会对结果产生影响。因为按照队列的取出顺序,数字 2 只能在所有的数字 1 被取出之后才能被取出,因此如果数字 1 如果在队列中,那么数字 2 必然也在队列中,使得数字 1 对结果没有影响。
按照上面的思路,我们可以设计这样的方法:从队列尾部插入元素时,我们可以提前取出队列中所有比这个元素小的元素,使得队列中只保留对结果有影响的数字。这样的方法等价于要求维持队列单调递减,即要保证每个元素的前面都没有比它小的元素。
代码
class MaxQueue:
def __init__(self):
self.que = []
self.que_max = []
def max_value(self) -> int:
if not self.que:
return -1
return self.que_max[0]
def push_back(self, value: int) -> None:
self.que.append(value)
while self.que_max and value > self.que_max[-1]:
self.que_max.pop()
self.que_max.append(value)
def pop_front(self) -> int:
if not self.que:
return -1
popvalue = self.que[0]
self.que.pop(0)
if popvalue==self.que_max[0]:
self.que_max.pop(0)
return popvalue