剑指 Offer 59 - II. 队列的最大值
题目描述
请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
class MaxQueue:
def __init__(self):
self.dq = collections.deque()
self.dq_m = collections.deque()
def max_value(self) -> int:
if not self.dq_m:
return -1
return self.dq_m[0]
def push_back(self, value: int) -> None:
self.dq.append(value)
while self.dq_m and self.dq_m[-1] < value:
self.dq_m.pop()
self.dq_m.append(value)
def pop_front(self) -> int:
if not self.dq:
return -1
if self.dq[0] == self.dq_m[0]:
self.dq_m.popleft()
return self.dq.popleft()
# Your MaxQueue object will be instantiated and called as such:
# obj = MaxQueue()
# param_1 = obj.max_value()
# obj.push_back(value)
# param_3 = obj.pop_front()