您的位置:首页 > 理论基础 > 数据结构算法

Python实现数据结构与算法——队列的最大值

2020-03-24 18:39 183 查看

题目描述:

请定义一个队列并实现函数 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路分析:

队列很好实现,主要是求最大值函数,以及均摊时间复杂度O(1)。解决办法,用辅助队列降序队列记录最大值,排序过程不要n次遍历就可以实现均摊时间复杂度为O(1)。辅助队列插入数字时,从后向前遍历,pop掉比当前值小的,然后加入当前数值。

代码如下:

from collections import deque

class MaxQueue:

def __init__(self):
self.queue = deque()
self.help_queue = deque()

def max_value(self) -> int:
return -1 if not self.queue else self.help_queue[0]

def push_back(self, value: int) -> None:
self.queue.append(value)
while self.help_queue and self.help_queue[-1] < value:
self.help_queue.pop()
self.help_queue.append(value)

def pop_front(self) -> int:
if not self.help_queue:
return -1
res = self.queue.popleft()
if res == self.help_queue[0]:
self.help_queue.popleft()
return res

测试结果:

总结:

觉得这道题难度不大,考虑好均摊时间复杂度O(1)就可以解决。同时分析时一定要记得,队列的FIFO的特点。

  • 点赞
  • 收藏
  • 分享
  • 文章举报
Mantana 发布了30 篇原创文章 · 获赞 11 · 访问量 2418 私信 关注
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: