leetcode算法学习(15)—— 队列的最大值

题目描述

请定义一个队列并实现函数 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]

方法一：暴力解决（时间复杂度不符合）

思路

直接实现一个普通的队列，查询最大值时遍历计算。

代码

import queue
class MaxQueue:

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

    def max_value(self) -> int:
        return max(self.deque) if self.deque else -1

    def push_back(self, value: int) -> None:
        self.deque.append(value)

    def pop_front(self) -> int:
        return self.deque.popleft() if self.deque else -1

方法二：添加辅助队列

思路

要想在O(1)时间内做到取出最大值，我们可以想到，能否用一个cur_max的变量，来记录并且比较每一次新入队的value，这个想法是极好的，但是如果队列是[4,3]这个样子，cur_max只会记下4是最大的，当调用一次pop_front()后，此时队列为[3]，而cur_max没有变化，所以单个变量记录最大值行不通。
进一步地我们可以想到，一个变量不行，那我直接用一个辅助队列记录值OK不OK呢？
答案是OK的。我们让辅助队列的数从大到小排列好，要找最大值直接返回辅助队列的头部即可，同时这也是O(1)时间复杂度的操作。

• 初始化两个队列：
  • 主队列：main_deque，正常出入队列
  • 辅助队列：sort_deque，对要入主队列的值进行非递增排序，队列头部记录主队列最大值
• sort_deque的如何排序问题
  • 队列的性质是先进先出，辅助队列在入队时，从辅助队列尾部依次取出比当前元素 value 小的元素。
• sort_deque的出队问题，以便最大值更新
  • 判断主队列要出队的元素是否等于辅助队列的头部，若等于将辅助队列的头部出队，即popleft()，以实时更新最大值。
• main_deque最大值问题
  • 直接返回辅助队列的第一个元素，即sort_deque[0]

代码

import queue
class MaxQueue:

    def __init__(self):
        self.main_deque=queue.deque()#主队列
        self.sort_deque=queue.deque()#辅助队列，维持一个非递增的队列，保证该队列头部一直是最大值

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

    def push_back(self, value: int) -> None:
        #主队列直接加入队列
        self.main_deque.append(value)
        #辅助队列在入队之前需要移除比该值小的元素
        while self.sort_deque and self.sort_deque[-1]<value:
            self.sort_deque.pop()
        self.sort_deque.append(value)
        
    def pop_front(self) -> int:
        if not self.main_deque:
            return -1
        #主队列出队
        res=self.main_deque.popleft()
        #辅助队列出队需要判断要出队的元素是否是最大值,若是，队列中的最大值发生变化，需要把原来的最大值出队
        if res==self.sort_deque[0]:
            self.sort_deque.popleft()
        return res