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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章