剑指Offer面试题59 - II. 队列的最大值

Posted 2021-03-06 galaxy-hao

题目

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

思路

代码

class MaxQueue {
    queue<int> que;
    deque<int> deq; 
public:
    MaxQueue() {

    }
    
    int max_value() {
        return deq.empty() ? -1 : deq.front();
    }
    
    void push_back(int value) {
        que.push(value);
        while (!deq.empty() && value > deq.back()) {
            deq.pop_back();
        } 
        deq.push_back(value);
    }
    
    int pop_front() {
        if (que.empty()) return -1;
        int res = que.front();
        que.pop();
        if (res == deq.front()) deq.pop_front();
        return res;        
    }
};