【剑指Offer】59-Ⅱ.队列的最大值

题目

请定义一个队列并实现函数 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 {
    public Queue q;
    public Deque d;

    public MaxQueue() {
        q = new LinkedList<>();
        d = new LinkedList<>();
    }
    
    public int max_value() {
        if (q.isEmpty()) {
            return -1;
        }

        return d.peekFirst();
    }
    
    public void push_back(int value) {
        q.offer(value);

        while (!d.isEmpty() && d.peekLast() < value) {
            d.pollLast();
        }
        d.offerLast(value);
    }
    
    public int pop_front() {
        if (q.isEmpty()) {
            return -1;
        }
        
        if (q.peek().equals(d.peekFirst())) {
            d.pollFirst();
        }

        return q.poll();
    }
}

/**
 * Your MaxQueue object will be instantiated and called as such:
 * MaxQueue obj = new MaxQueue();
 * int param_1 = obj.max_value();
 * obj.push_back(value);
 * int param_3 = obj.pop_front();
 */

总结

这题和滑动窗口挺类似的，也是维护了一个辅助双端队列来记录队列中的最大值。

详情见：【LeetCode】239.滑动窗口最大值

这里有一个坑需要注意：

pop_front中如果使用q.peek() == d.peekFirst()做判定条件，是不对的。

用q.peek().equals(d.peekFirst())作判定条件，才是对的。

因为队列peek()或者peekFirst()返回的对象不是整型，而是Object，所以不能用==，而应该用equals。