题目
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路
本题要找到右边第一个比自己大的元素
单调栈的使用场景:
通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时就要想到可以用单调栈了
单调栈要注意的几点:
- 单调栈存放的是元素的下标
- 明确单调栈的维护顺序(递增/递减)
- 判断当前遍历的元素和栈顶元素的关系
①当前元素 < 栈顶元素
②当前元素 == 栈顶元素
③当前元素 > 栈顶元素
在本题中,因为要找到右边第一个比自己大的元素
所以维护递减栈,即在栈中,后入栈的元素总比先入栈的元素小,本题中就是从栈底到栈顶(头)的下标对应的温度列表中的温度依次递减,这样的话可以保证,如果最后有下标在单调栈里,则表示尚未找到下一次温度更高的下标,因为如果有更高的温度,那么就会将栈顶元素出栈
- 比对当前元素与栈顶元素的大小
(1)若当前元素 < 栈顶元素:入栈
(2)若当前元素 > 栈顶元素:弹出栈顶元素,将当前元素入栈,并记录两者下标差值即为所求天数
算法思想:
正向遍历温度列表,对于温度列表中的每个元素 temperatures[i]
:
(1)如果栈为空,则直接将 i
进栈
(2)如果栈不为空,则比较栈顶元素 prevIndex
对应的温度temperatures[prevIndex]
和当前温度 temperatures[i]
;如果 temperatures[i] > temperatures[prevIndex]
(当前元素>栈顶元素,比较的是下标对应的元素),则将 prevIndex
移除(出栈的是下标),并将 prevIndex
对应的等待天数赋为 i - prevIndex
(两者的下标差),重复上述操作直到栈为空或者栈顶元素对应的温度小于等于当前温度,然后将 i
进栈
java代码如下:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int len = temperatures.length;
int[] ans = new int[len];
Deque<Integer> stack = new LinkedList<Integer>();//Stack<Integer> stack = new Stack<>()也可以
for (int i = 0; i < len; i++) {
int temperature = temperatures[i];
while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {//因为单调栈存的是元素下标,所以temperatures[stack.peek()]表示的是当前stack.peek()的下标对应的温度
int prevIndex = stack.pop();//将当前栈顶元素出栈
ans[prevIndex] = i - prevIndex;//计算两个下标天数差,并存入结果
}
stack.push(i);
}
return ans;
}
}