题目一:队内最大值
链接: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
例1
输入:
fun = ["MaxQueue","push_back","push_back","max_value","pop_front","max_value"]
num = [[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]
例2
输入:
fun = ["MaxQueue","pop_front","max_value"]
num = [[],[],[]]
输出: [null,-1,-1]
fun数组代表的是每次执行的函数,
num数组代表的是每次执行的函数所需要传入的参数值。
Deque模块:
Deque是Python库collections中的一项,它可以在序列前后都执行添加或删除.
.append()—右插入
.appendleft()—左插入
.pop()—右删除
.popleft()—左删除
思路:
1、先创建两个队列,分别为que和max_que,que队列用来插入数据,max_que用来存储对内最大值(单调递减);
2、max_que按照每次插入的数value进行比较,当max_que[-1] 小于插入value的时候,移除max_que[-1],直到max_que[-1] > value或者max_que队列没有数据停止;
3、如果移除的数据是对内的最大值,那么max_que中也要相对应移除
class MaxQueue(object):
def __init__(self):
from collections import deque
self.que = deque()
self.max_que = deque()
def max_value(self):
"""
:rtype: int
"""
return self.max_que[0] if self.max_que else -1
def push_back(self, value):
"""
:type value: int
:rtype: None
"""
while self.max_que and self.max_que[-1] < value:
self.max_que.pop()
self.max_que.append(value)
self.que.appendleft(value)
def pop_front(self):
"""
:rtype: int
"""
if not self.que:
return -1
num = self.que.pop()
if num == self.max_que[0]:
self.max_que.popleft()
return num