essay
hot100-每日温度
#算法
hot100——每日温度
给定一个整数数组 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]
提示:
1 <= temperatures.length <=
30 <= temperatures[i] <= 100
解法
func dailyTemperatures(temperatures []int) []int {
n := len(temperatures)
res := make([]int, n) // 结果数组,默认值0(一定要使用make初始化,因为后面直接对res[i]赋值了!)
stack := make([]int, 0) // 单调栈,存储索引,栈顶对应的温度最小
for i := 0; i < n; i++ {
// 当前温度比栈顶索引对应的温度高,说明找到了下一个更高温度
for len(stack) > 0 && temperatures[i] > temperatures[stack[len(stack)-1]] {
top := stack[len(stack)-1] // 栈顶索引
stack = stack[:len(stack)-1] // 弹出
res[top] = i - top // 计算天数差
}
// 将当前索引入栈
stack = append(stack, i)
}
// 栈中剩余的元素没有找到更高温度,res中默认0,无需处理
return res
}