题目
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
分析
这个题第一个想法是双指针,即从左侧向右扫描,left指针和right指针,遇到right指向的元素比left大的,就计算隔的天数,然后left前进一步,right回到left+1处,判断并继续向前扫描。不过提交时候超时了。。。这其实相当于双层循环,暴力解法。
题解里,更为巧妙的解法是,单调栈。这是一种专门针对前后左右的,比如接雨水的问题,也是这种解法。
在栈里存储的是索引,然后遍历整个温度,遇到温度比栈里的大的,就计算两个索引之间的距离,并对结果集赋值,也就是栈里这个索引所在的结果,然后将其弹出。继续拿栈顶与当前温度比较。
解法
func dailyTemperatures(T []int) []int {
// 当前栈顶的是73,压入一个74,1,
// 当前栈顶的是74,压入一个75,1,
// 当前栈顶是75,压入一个71,+1,当前计算的是75。
// 当前栈顶的是71,压入一个69,+1
// 当前栈顶的是69,压入一个72,+1,
// 当前栈顶的是72,压入一个76,+1,结束,day=4,
// 栈弹出4-1个,弹3个出来,76,72,69,栈顶是71,
// 当前栈顶是71,压入一个69,+1,
// 当前栈顶是69,压入一个72,+1,结束,day=2
// 栈弹出2-1,弹一个出来72,栈顶是69
// 当前栈顶是69,压入一个,
// 遇到结尾,没有找到大的,day=0,弹出n-1个来。
// 双指针超时
// var result []int
// for i, j:=0, 1; j < len(T); {
// if T[i] < T[j] {
// result = append(result, j-i)
// i++
// j=i+1
// } else {
// if j == len(T) - 1 {
// result = append(result, 0)
// i++
// j=i+1
// } else {
// j++
// }
// }
// }
// // 最后还要补上一个最后一位没被比较的0
// result = append(result, 0)
var stack []int
var result = make([]int, len(T))
for i := 0; i < len(T); {
// 当前温度大于栈顶索引对应的问题,则对应的栈顶索引出等待天数为i- stack.peek()
if len(stack) > 0 && T[i] > T[stack[len(stack)-1]] {
result[stack[len(stack)-1]] = i - stack[len(stack)-1]
stack = stack[:len(stack)-1]
} else {
stack = append(stack, i)
i++
}
// fmt.Printf("stack: %v, t: %d, index : %d, \n", stack, T[i], i)
}
return result
}