题目背景:
根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数
一看这题简单啊,立马就想到两层for循环,找到比当前值立马结束。于是:
public static int[] dailyTemperatures(int[] T) {
int[] r = new int [T.length];
for (int i = 0;i < T.length; i++){
for (int j = i + 1; j < T.length; j++){
if (T[i] < T[j]){
r[i] = j -i;
break;
}
}
}
return r;
}
结果: 微微飘过,后来一想这个得跟栈有关啊 ,一看答案,卧槽,牛逼,头发秃了我也想不到啊。
官方的解释:
我们需要找到比当前 T[i] 温度更高的位置,那么必须要记录哪些信息?
我们试着找到 T[0] 过后温度升高的位置。如果知道 T[10]=50,则 T[20]=50 是无效信息,因为 T[i] 在 T[20] 以前已经到达了 50。如果 t[20]=100 将是有用的信息,因为如果 t[0]=80,那么 T[20] 将有可能是它的下一个温度升高的位置,而 T[10] 则不可能是。
因此,我们需要记住一个索引的列表,索引代表的温度严格递增。我们可以利用栈来实现这样的效果。
算法:
我们用栈记录索引,满足 T[stack[-1]] < T[stack[-2]] < …,其中 stack[-1] 是栈的顶部,stack[-2] 是从顶部开始的第二个元素,依此类推;我们将在处理每个 T[i] 时保持 stack[-1] > stack[-2] > …。
我们通过当前温度和栈顶索引所代表的温度比较来找到温度升高的位置。
举个例子:我们反向遍历处理 t=[73,74,75,71,69,72,76,73] ,通过看栈元素的变化来理解是如何工作的。为了清楚 stack 只包含索引 i,但是将把 T[i] 的值写在旁边的括号中,例如 0 (73)。
当 i = 7,stack = [7 (73)]。ans[i] = 0。
当 i = 6,stack = [6 (76)]。ans[i] = 0。
当 i = 5,stack = [5 (72), 6 (76)]。ans[i] = 1。
当 i = 4,stack = [4 (69), 5 (72), 6 (76)]。ans[i] = 1。
当 i = 3,stack = [3 (71), 5 (72), 6 (76)]。ans[i] = 2。
当 i = 2,stack = [2 (75), 6 (76)]。ans[i] = 4。
当 i = 1,stack = [1 (74), 2 (75), 6 (76)]。ans[i] = 1。
当 i = 0,stack = [0 (73), 1 (74), 2 (75), 6 (76)]。ans[i] = 1。
作者:LeetCode
看着有点蒙,后来想明白来了,结果要的是比当前数值的大的第一个数,和当前索引的差值。
所以:
第一件事倒序,为啥倒序,一遍解决问题,降低复杂度。
第二,保持栈内的元素的顺序保持递增:栈低到栈顶 由大到小,存储的是数组中的索引(结果要的就是索引差值)。当前元素大于栈定元素弹出元素,直到不弹出,这时栈要是不空的话,栈定元素就是当前温度最近的升温日期。二者差值即为ans[i];故代码如下:
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] ans = new int[T.length];
Stack<Integer> stack = new Stack();
for (int i = T.length - 1; i >= 0; --i) {
while (!stack.isEmpty() && T[i] >= T[stack.peek()]) {
stack.pop();
}
//栈内的元素的顺序保持递增:栈低到栈顶 由大到小,存储的是数组中的索引
ans[i] = stack.isEmpty() ? 0 : stack.peek() - i;
stack.push(i);
}
return ans;
}
}