leetcode739. 每日温度
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/daily-temperatures
题目描述
给定一个整数数组 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]
解法一 暴力解
循环嵌套去判断,这是最容易想到的方法,但超出时间限制,提交不通过
/**
* 暴力解
* @param temperatures
* @return
*/
public int[] dailyTemperatures2(int[] temperatures) {
if (temperatures.length < 2){
return new int[1];
}
int N = temperatures.length;
int[]ans = new int[N];
for (int i = 0; i < N;i++){
for (int j = i + 1;j < N;j++){
if (temperatures[j] > temperatures[i]){
ans[i] = j - i;
break;
}
}
}
return ans;
}
解法二 单调栈
为了方便,我们令 temperatures 为 ts。
抽象题意为 : 求解给定序列中每个位置(右边)最近一个比其大的位置,可使用「单调栈」来进行求解。
具体的,我们可以从前往后处理所有的
,使用某类容器装载我们所有的「待更新」的位置(下标),假设当前处理到的是
若其比容器内的任意位置(下标)对应温度要低,其必然不能更新任何位置(下标),将其也加入容器尾部(此时我们发现,若有一个新的位置(下标)加入容器,其必然是当前所有待更新位置(下标)中的温度最低的,即容器内的温度单调递减);
若其价格高于容器内的任一位置(下标)对应温度,其能够更新容器位置(下标)的答案,并且由于我们容器满足单调递减特性,我们必然能够从尾部开始取出待更新位置来进行更新答案,直到处理完成或遇到第一个无法更新位置。
由于我们需要往尾部添加和取出元素,因此我们可以用数组来优化栈结构。
/**
* 单调栈
* @param temperatures
* @return
*/
public int[] dailyTemperatures(int[] temperatures) {
if (temperatures.length < 2){
return new int[1];
}
int N = temperatures.length;
int[]ans = new int[N];
//用数组来优化栈
int[]stack = new int[N];
int stackSize = 0;
for (int i = 0; i < N;i++){
while (stackSize != 0 && temperatures[stack[stackSize - 1]] < temperatures[i]){
int cur = stack[--stackSize];
ans[cur] = i - cur;
}
stack[stackSize++] = i;
}
return ans;
}