2021-12-23:每日温度。 请根据每日 气温 列表 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]
力扣739。
答案2021-12-23:
时间紧。具体见代码。
代码用golang编写。代码如下:
package main
import "fmt"
func main() {
arr := []int{73, 74, 75, 71, 69, 72, 76, 73}
ret := dailyTemperatures(arr)
fmt.Println(ret)
}
func dailyTemperatures(arr []int) []int {
if len(arr) == 0 {
return []int{}
}
N := len(arr)
ans := make([]int, N)
stack := make([][]int, 0)
for i := 0; i < N; i++ {
for len(stack) > 0 && arr[stack[len(stack)-1][0]] < arr[i] {
popIs := stack[len(stack)-1]
stack = stack[0 : len(stack)-1]
for _, popi := range popIs {
ans[popi] = i - popi
}
}
if len(stack) > 0 && arr[stack[len(stack)-1][0]] < arr[i] {
stack[len(stack)-1] = append(stack[len(stack)-1], i)
} else {
list := make([]int, 0)
list = append(list, i)
stack = append(stack, list)
}
}
return ans
}
执行结果如下:
左神java代码
|