请定义一个队列并实现函数 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
import queue
class MaxQueue:
def __init__(self):
# 这个的作用是列表的首位是我们想要的最大值
self.deque = queue.deque()
# 这个的作用是正常的入队,出队的
self.queue = queue.Queue()
# 就是在正常列表弹出,推入的过程中,使得求最大值的均摊时间复杂度更小
def max_value(self) -> int:
# 如果队列不为空,则直接返回队列首
# print(self.deque)
return self.deque[0] if self.deque else -1
def push_back(self, value: int) -> None:
# 如果队列不为空,且队列的首元素小于要添加的值
while self.deque and self.deque[-1] < value:
# 则将最后一个数字弹出,那么队列中就会存在现在最大的值,为了等会求最大值做准备
self.deque.pop()
# 双端队列的尾部添加一个元素
# print(value)
self.deque.append(value)
self.queue.put(value)
def pop_front(self) -> int:
# 如果双端队列为空,直接返回-1
if not self.deque:
return -1
# 从原始队列中获取一个值
ans = self.queue.get()
# print(ans)
# 如果原始队列等于双向队列的头部,则双向队列的左侧进行弹出
if ans == self.deque[0]:
self.deque.popleft()
# 弹出原始队列
return ans
# 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()