劍指 Offer 59 - II. 佇列的最大值

題目

連結

https://leetcode-cn.com/problems/dui-lie-de-zui-da-zhi-lcof/solution/mian-shi-ti-59-ii-dui-lie-de-zui-da-zhi-by-leetcod/

問題描述

請定義一個佇列並實現函式 max_value 得到佇列裡的最大值，要求函式max_value、push_back 和 pop_front 的均攤時間複雜度都是O(1)。

若佇列為空，pop_front 和 max_value 需要返回 -1

示例

輸入:
["MaxQueue","push_back","push_back","max_value","pop_front","max_value"]
[[],[1],[2],[],[],[]]
輸出: [null,null,null,2,1,2]

提示

1 <= push_back,pop_front,max_value的總運算元 <= 10000
1 <= value <= 10^5

思路

這裡維護兩個佇列，一個存放資料，一個存放最大值，當入棧時，把存放最大值的數裡面，小於新的數的全部取出。

複雜度分析

時間複雜度 O(1)
空間複雜度 O(n)

程式碼

Java

    Queue<Integer> q;
    Deque<Integer> d;

    public JZ59II() {
        q = new LinkedList<Integer>();
        d = new LinkedList<Integer>();
    }

    public int max_value() {
        if (d.isEmpty()) {
            return -1;
        }
        return d.peekFirst();
    }

    public void push_back(int value) {
        while (!d.isEmpty() && d.peekLast() < value) {
            d.pollLast();
        }
        d.offerLast(value);
        q.offer(value);
    }

    public int pop_front() {
        if (q.isEmpty()) {
            return -1;
        }
        int ans = q.poll();
        if (ans == d.peekFirst()) {
            d.pollFirst();
        }
        return ans;
    }