Loading

Medium | 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] 范围内的整数。

解题思路

方法一:单调栈

使用一个单调递减栈, 每次遇到一个比栈顶元素大的元素的数之和, 都可以计算栈内的比当前元素的小的元素的等待的天数。当遍历完成后, 栈内如果还有元素, 则说明, 栈内的这些元素的后面, 没有比他大的数。则填充为0。

public int[] dailyTemperatures(int[] T) {
    Stack<Integer> stack = new Stack<>();
	// 先将下标为0的元素压栈, 这里保存下标即可, 并不保存实际的元素值, 方便计算等待天数
    stack.push(0);
    int[] res = new int[T.length];
    int lastIndex;
    for (int i = 1; i < T.length; i++) {
        // 每次压栈过程当中, 将栈顶的比当前小的元素出栈, 并计算等待天数
        while (!stack.empty() && (T[i] > T[(lastIndex = stack.peek())])) {
            res[lastIndex] = i - lastIndex;
            stack.pop();
        }
        stack.push(i);
    }
    // 所有元素遍历完成后, 栈内的数字是最后的连续的单调递减的下标。设置等待天数为0
    int index = stack.get(0);
    for (int i = index; i < res.length; i++) {
        res[index] = 0;
    }
    return  res;
}
posted @ 2021-01-31 11:42  反身而诚、  阅读(50)  评论(0编辑  收藏  举报