leetcode *739. 每日溫度(單調棧)

【題目】*739. 每日溫度

根據每日 氣溫 列表,請重新生成一個列表,對應位置的輸出是需要再等待多久溫度纔會升高超過該日的天數。如果之後都不會升高,請在該位置用 0 來代替。
例如,給定一個列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的輸出應該是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:氣溫 列表長度的範圍是 [1, 30000]。每個氣溫的值的均爲華氏度,都是在 [30, 100] 範圍內的整數。

【解題思路1】暴力法

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int[] res = new int[T.length];
        for (int i = 0; i < T.length; i++) {            
            int j = i + 1;
            while (j < T.length && T[i] >= T[j]) {                
                j++;  
            }
            res[i] = j < T.length && T[i] < T[j] ? j - i : 0;
        }
        return res;
    }
}

優化的暴力法,來自於官方題解評論區

class Solution {
    /** 暴力優化 —— 自底向上 */
    public int[] dailyTemperatures(int[] T) {
        int n = T.length;
        int[] res = new int[n];
        for (int i = n - 2; i >= 0; --i) {
            // j += res[j] ==> 當T[i]<T[j]時,往後遍歷比T[j]溫度高的情況
            for (int j = i + 1; j < n; j += res[j]) {
                // 找到溫度高的一天
                if (T[i] < T[j]) {
                    res[i] = j - i;
                    break;
                }
                // res[j] == 0說明第j天以後沒有比第j天更高的溫度了
                else if (res[j] == 0) {
                    res[i] = 0;
                    break;
                }
            }
        }
        return res;
    }
}

【解題思路2】單調棧

單調棧動畫
維護一個存儲下標的單調棧,從棧底到棧頂的下標對應的溫度列表中的溫度依次遞減。如果一個下標在單調棧裏,則表示尚未找到下一次溫度更高的下標。

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int[] res = new int[T.length];
        Deque<Integer> stack = new LinkedList<Integer>();
        for(int i = 0; i < T.length; i++) {
            while(!stack.isEmpty() && T[i] > T[stack.peek()]) {
                res[stack.peek()] = i - stack.pop();
            }
            stack.push(i);
        }
        return res;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章