题目描述:
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
示例2:
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]
示例3:
输入: temperatures = [30,60,90]
输出: [1,1,0]
思路:从后往前遍历,若后一个元素大于当前元素(T[i+1]>T[i]),则ans[i]=1。若小于,则与后一个元素的上一级比较(T[ans[i+1]+i+1]),直到有大于T[i]的元素出现。特殊情况是相等的时候,要判断相等元素的ans是否为0,若为0则说明该元素的右侧没有更大元素,则ans[i]同样为0;若不为0,则更新距离即可。While循环时注意break条件,不然容易死循环。评论区里另一种用单调栈的方法,我的复现代码版列在了第二种方法中
代码1:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int len = temperatures.length;
int[] ans = new int[len];
ans[len-1] = 0;
for (int i = len - 2; i >= 0; i--) {
if (temperatures[i+1] > temperatures[i]) {
ans[i] = 1;
} else if (temperatures[i+1] < temperatures[i]){
int p = i+1+ans[i+1];
while (p < len) {
if (temperatures[p] > temperatures[i]) {
ans[i] = p - i;
break;
} else if (temperatures[p] < temperatures[i]) {
if (ans[p] == 0) break; //这里一定不能漏!
else p = p+ans[p];
} else { //相等时
if (ans[p] != 0) { //相等元素的后边位置存在比它大的
ans[i] = ans[p] + p - i;
} else { //相等元素在它之后没有小于它的值
ans[i] = 0;
}
break;
}
}
} else { //这边也要判断
if(ans[i+1] != 0) ans[i] = ans[i+1] + 1;
else ans[i] = 0;
}
}
return ans;
}
}
代码2(Stack):
class Solution {
public int[] dailyTemperatures(int[] T) {
Stack<Integer> stk = new Stack<Integer>();
int[] ans = new int[T.length];
for (int i = 0; i < T.length; i++) {
while (!stk.empty() && T[stk.peek()] < T[i]) {
ans[stk.peek()] = i - stk.peek();
stk.pop();
}
stk.push(i); //因为无论与栈顶元素的比较结果如何,当前i位置的元素肯定要进栈。
}
return ans;
}
}
代码2虽然短,但耗时较长。