题目
请根据每日 气温
列表 temperatures
,重新生成一个列表,要求其对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 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]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
注意:本题与主站 739 题相同: 力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
题解
思路一:暴力解法,因为温度是从30-100.使用一个sum数组来存储所有出现的温度,对应的距离当前元素最近的下标。从后向前遍历原始数组。sum[i],i表示温度,sum[i]表示在原数组中的下标。因为是后序遍历的,因此一定是出现在后面的更高温度。
代码:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] sum = new int[101];
int[] ans=new int[temperatures.length];
Arrays.fill(sum,Integer.MAX_VALUE);
for(int i=temperatures.length-1;i>=0;i--) {
int index=Integer.MAX_VALUE;
for(int j=temperatures[i]+1;j<101;j++) {
if(sum[j]<index) {
index=sum[j];
}
}
if(index<Integer.MAX_VALUE)
ans[i]=index-i;//一定要记得-i,因为是从i开始算第多少个温度更高
sum[temperatures[i]]=i;
}
return ans;
}
}
思路二:单调栈思想,栈中存储下标,栈中数据所代表的温度从栈底到栈顶是从高到低的,从前向后遍历原始数组。栈不空时,当有温度元素大于栈顶元素时,取出栈顶元素,更新ans[栈内index](代表的是还需要几天才升高)=当前index-栈内index;温度小于栈顶或者栈空时,直接入栈,最后将所有的栈内残留ans[index]=0。
代码:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] ans = new int[temperatures.length];
Deque<Integer> stack = new ArrayDeque<Integer>();
for (int i = 0; i < length; i++) {
int temperature = temperatures[i];
while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
int prevIndex = stack.pop();
ans[prevIndex] = i - prevIndex;
}
stack.push(i);
}
return ans;
}
}