0739.每日温度
描述
根据每日 气温
列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0
来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
提示
气温
列表长度的范围是 [1, 30000]
。每个气温的值的都是 [30, 100]
范围内的整数。
题解
超时示例
- 从后往前遍历,当前元素为
i
- 遍历第一个到第
i
个元素,当前元素为j
- 如果第
i
元素大于j
元素,result[j]
值为i-j
- 遍历第一个到第
时间复杂度为O(n^2)
public int[] dailyTemperatures(int[] T) {
int[] result = new int[T.length];
for (int i=T.length-1;i>=0;i--){
for (int j=i-1;j>=0;j--){
if (T[i] > T[j])
result[j] = i-j;
}
}
return result;
}
正确解法
使用stack
用于存储尚未找到最近升温天气的元素位置
栈顶元素为当前栈中最小值
public 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.empty()){
int j = stack.peek();
if (T[i] > T[j]){
result[j] = i-stack.pop();
} else {
break;
}
}
stack.push(i);
}
return result;
}