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;
}