题目描述:
根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:
气温 列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。
解题思路:
首先最容易想到的是,用双循环来做。遍历整个数组,找到第一个比当前元素大的元素,记录下元素之间的length。
代码实现:
public static int[] dailyTemperatures(int[] T) {
int[] result = new int[T.length];
for (int i = 0; i < T.length; i++) {
int step = 0;
for (int j = i+1; j < T.length; j++) {
step++;
if (T[j] > T[i]) {
result[i] = step;
break;
}
continue;
}
}
return result;
}
运行结果:
用时410ms,只击败了21%的用户。而且这道题是中等难度,显然此解法并非迎合出题人的用意所在。还有没有更好的办法呢?
考虑到可以利用栈的性质来做。此方法只遍历一次数组即可完成。主要思路就是:用一个栈来记录数组元素的下标。为什么记录下标呢?因为题目中要求的天数即为数组元素下标的差值,并且有了数组下标,也非常方便可以取到数组元素中的值。当当前元素的值大于栈顶元素时,则直接记录两元素下标差值,并且将栈顶元素出栈;若当前元素小于栈顶元素时,则将当前元素入栈。用文字描述也不清晰,下面是算法具体步骤的描述,参考这位博主的:
https://blog.csdn.net/qq_39360985/article/details/83052418
代码实现:
public static int[] dailyTemperatures(int[] T) {
Stack<Integer> stack = new Stack<>();
int[] result = new int[T.length];
for (int i = 0; i < T.length; i++) {
while(!stack.isEmpty() && T[stack.peek()] < T[i]){
int peek = stack.peek();
result[peek] = i - peek;
stack.pop();
}
stack.push(i);
}
return result;
}
运行结果:
效果显著。