739. 每日温度 中等

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

package main

import "fmt"

func main() {
    fmt.Println(dailyTemperatures([]int{73, 74, 75, 71, 69, 72, 76, 73}))
}

// 十分有栈味道的题
// 弄清楚栈在题中如何使用,灵活结合 pop push 等操作解决
func dailyTemperatures(temperatures []int) []int {
    res := make([]int, len(temperatures))
    var s Stack // 索引栈
    for i, t := range temperatures {
        for !s.IsEmpty() {
            peek := s.Peek()
            if t > temperatures[peek] { // 遇到大数,则把积压的小数全部 pop 出来
                res[peek] = i - peek
                s.Pop()
            } else {
                break
            }
        }
        s.Push(i)
    }
    return res
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng