题目要求:
根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路1:
第一次看到这个题的时候,想着先从简单的办法写起:
那么就来了,就是外层for循环 i 遍历数组,内层循环 从 j = i + 1开始遍历,找寻大于外层循环元素的值,
- 找到了 结果数组 res[i] = j - i
- 到了数组末尾了还没找到,res[i] = 0
代码:
public static int[] dailyTemperaturesDemo(int[] T){
int len = T.length;
int[] res = new int[len];
for(int i = 0; i < len;i++) {
int j = i + 1;
while (j < len && T[j] <= T[i]){
j++;
}
if(j == len)
res[i] = 0;
else
res[i] = j - i;
}
return res;
}
但是后面想着用别的办法解的时候,却迟迟想不出来,所以看了题解,发现正确用对数据结构,真的很重要。
思路2:
很明显,可以看出,前面的方法,有多次重复计算的过程。
比如数组: {73,74,75,71,69,72,76,73} 当遍历到75时, 后面几个数都是小于它的,内层循环肯定会遍历,直到找到76这个数,内层循环才会结束。 然后外层循环遍历到71, 它的内层循环又会重复计算69与71的大小关系。
所以优化就是为了找到避免重复计算的方法。
递减栈解法:
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是 递减栈 ,所以需要取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离。
继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来。
作者:MisterBooo
链接
代码:
public static int[] dailyTemperaturesDemo2(int[] T) {
if(T == null || T.length == 0)
return null;
int ans[] = new int[T.length];
Stack<Integer> stack = new Stack<>();
int len = T.length;
for(int i = 0; i < len;i++) {
//遇到比当前栈顶元素大的T[i] 弹出栈顶元素,然后将结果数组ans的值更改,如果T[i] < 当前栈顶元素, 则入栈
while (!stack.isEmpty() && T[i] > T[stack.peek()]) {
int top = stack.pop();
ans[top] = i - top;
}
stack.push(i);
}
return ans;
}
此解法,leecode用时18ms,而第一个解法用时925ms 孰强孰弱,一眼便知。 所以说选择一些好的数据结构解题,真的会事半功倍。
加油,路还长~
还有一些类似题:
- 求解算术表达式的结果(LeetCode 224、227、772、770)
- 求解直方图里最大的矩形区域(LeetCode 84)