Skip to the content.

leetcode [面试题59 - II] 队列的最大值


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

来自题解:

两个队列,一个存储元素,一个是单调递减栈。注意pop的时候如果两个队列首元素相等才pop另一个递减栈

class MaxQueue:

    def __init__(self):
        self.q = collections.deque()#数据队列
        self.p = collections.deque()#单调递减栈(准确说是非增)

    def max_value(self) -> int:
        return not self.q and -1 or self.p[0] #数据队列为空,返回-1,否则返回单调栈第一个元素

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

    def pop_front(self) -> int:
        if not self.q:
            return -1
        self.q[0] == self.p[0] and self.p.popleft() #数据队列元素为最大值时才出栈单调栈元素
        return self.q.popleft()