739、每日温度
根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
提示:气温 列表长度的范围是 [1, 30000]
。每个气温的值的均为华氏度,都是在 [30, 100]
范围内的整数。
题解
1、暴力
- 原始暴力
class Solution {
public int[] dailyTemperatures(int[] T) {
// 后面的 最近的 比自己大的 距离
int[] res = new int[T.length];
for (int i = 0;i < T.length;i++) {
for (int j = i+1;j < T.length;j++) {
if (T[j] > T[i]) {
res[i] = j - i;
break;
}
}
}
return res;
}
}
- 高级暴力
class Solution {
public int[] dailyTemperatures(int[] T) {
int length = T.length;
int[] ans = new int[length];
int[] next = new int[101];
Arrays.fill(next, Integer.MAX_VALUE);
// 从后往前
for (int i = length - 1; i >= 0; --i) {
// 最近的较大值
int warmerIndex = Integer.MAX_VALUE;
for (int t = T[i] + 1; t <= 100; ++t) {
if (next[t] < warmerIndex) {
warmerIndex = next[t];
}
}
// 找到最近较大值
if (warmerIndex < Integer.MAX_VALUE) {
ans[i] = warmerIndex - i;
}
// 记录T[i]最近的下标
next[T[i]] = i;
}
return ans;
}
}
// 链接:https://leetcode-cn.com/problems/daily-temperatures/solution/mei-ri-wen-du-by-leetcode-solution/
- 巧妙暴力
class Solution {
/**
* 根据题意,从最后一天推到第一天,这样会简单很多。因为最后一天显然不会再有升高的可能,结果直接为0。
* 再看倒数第二天的温度,如果比倒数第一天低,那么答案显然为1,如果比倒数第一天高,又因为倒数第一天
* 对应的结果为0,即表示之后不会再升高,所以倒数第二天的结果也应该为0。
* 自此我们容易观察出规律,要求出第i天对应的结果,只需要知道第i+1天对应的结果就可以:
* - 若T[i] < T[i+1],那么res[i]=1;
* - 若T[i] > T[i+1]
* - res[i+1]=0,那么res[i]=0;
* - res[i+1]!=0,那就比较T[i]和T[i+1+res[i+1]](即将第i天的温度与比第i+1天大的那天的温度进行比较)
*/
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
res[T.length - 1] = 0;
// 从后往前
for (int i = T.length - 2; i >= 0; i--) {
// 下标跳跃,j += res[j] 跳跃至T[j]最近的较大值
for (int j = i + 1; j < T.length; j += res[j]) {
if (T[i] < T[j]) {// 符合,直接存储
res[i] = j - i;
break;
} else if (res[j] == 0) {// T[j] 后面没有更大的了
res[i] = 0;
break;
}
}
}
return res;
}
}
2、单调栈
可以维护一个存储下标的单调栈,从栈底到栈顶的下标对应的温度列表中的温度依次递减。如果一个下标在单调栈里,则表示尚未找到下一次温度更高的下标。
先正向遍历,对于温度列表中的每个元素 T[i],如果栈为空,则直接将 i 进栈,如果栈不为空,则比较栈顶元素 prevIndex 对应的温度 T[prevIndex] 和当前温度 T[i],如果 T[i] > T[prevIndex],则将 prevIndex 移除,并将 prevIndex 对应的等待天数赋为 i - prevIndex,重复上述操作直到栈为空或者栈顶元素对应的温度小于等于当前温度,然后将 i 进栈。
为什么可以在弹栈的时候更新 ans[prevIndex] 呢?因为在这种情况下,即将进栈的 i 对应的 T[i] 一定是 T[prevIndex] 右边第一个比它大的元素,试想如果 prevIndex 和 i 有比它大的元素,假设下标为 j,那么 prevIndex 一定会在下标 j 的那一轮被弹掉。
由于单调栈满足从栈底到栈顶元素对应的温度递减,因此每次有元素进栈时,会将温度更低的元素全部移除,并更新出栈元素对应的等待天数,这样可以确保等待天数一定是最小的。
class Solution {
public int[] dailyTemperatures(int[] T) {
int len = T.length;
int[] ans = new int[len];
Deque<Integer> stack = new LinkedList<>();
for (int i = 0;i < len;i++) {
int temp = T[i];
// 当前元素比栈顶元素大,找到最近的较大值
while(!stack.isEmpty() && temp > T[stack.peek()]) {
int prevIndex = stack.pop();
// 存储结果
ans[prevIndex] = i - prevIndex;
}
stack.push(i);
}
return ans;
}
}
转载请注明来源,欢迎对文章中的引用来源进行考证,欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论,也可以邮件至 1056615746@qq.com