739. 每日温度
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
方法1:暴力解法
双重循环遍历即可
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
for (int i = 0; i < T.length - 1; i++) {
for (int j = i + 1; j < T.length; j++) {
if (T[j] > T[i]) {
res[i] = j - i;
break;
}
}
}
return res;
}
方法2:单调栈
维护一个单调递减栈,栈中存放气温数组的下标。所谓单调递减栈,就是栈的底部到顶部是单调递减的。从头遍历气温数组T,若栈为空,直接将数组下标入栈;若栈不为空,并且当前遍历到的下标对应的温度T[i] 大于栈顶下标对应的温度,那么说明这个温度是栈顶元素的下一个更高的温度,将栈顶元素弹出,两者下标的差值即为所求的天数,注意这个判断应该用while循环。
时间复杂度O(N)
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
//单调递减栈
Deque<Integer> stack = new ArrayDeque<>();
for (int i = 0; i < T.length; i++) {
while (!stack.isEmpty() && T[i] > T[stack.peekLast()]) {
int top = stack.peekLast();
res[stack.pollLast()] = i - top;
}
stack.addLast(i);
}
return res;
}