大家好,我是木川
一、题目描述
给定一个整数数组 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 <= 105
30 <= temperatures[i] <= 100
二、题解
一)暴力算法
思路:先看下最简单的思路,暴力算法肯定很耗时,肯定不会通过,这里写出来只是了解下思路
1、 第一层循环遍历数组元素
2、第二层循环遍历当前元素后面的元素,如果碰到比当前元素大,则结束
代码:
func dailyTemperatures(temperatures []int) []int {
res := make([]int, len(temperatures))
for i := 0; i < len(temperatures); i++ {
find := false
for j := i+1; j < len(temperatures); j++ {
res[i] ++
if temperatures[j] > temperatures[i] {
find = true
break
}
}
// 没找到比当前元素的大
if !find {
res[i] = 0
}
}
return res
}
时间复杂度:O(n ^ 2)
空间复杂度:O(n)
二)单调栈栈
单调栈:栈中元素要么单调递减,要么单调递增,本题使用递减栈
思路:
1、遍历整个数组,如果栈不空,且当前数字大于栈顶元素则取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离
2、继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来
代码:
func dailyTemperatures(temperatures []int) []int {
res := make([]int, len(temperatures))
stack := make([]int, 0)
for i := 0; i < len(temperatures); i++ {
// 当前元素 > 栈顶元素
for (len(stack) > 0 && temperatures[i] > temperatures[stack[len(stack)-1]]) {
// 栈顶元素对应的下标(之前访问的元素)
// 当前下标 - 栈顶元素下标 = 第一个大于它的数的距离
idx := stack[len(stack)-1]
res[idx] = i - idx
// 出栈操作:前一个元素下标出栈
stack = stack[0:len(stack)-1]
}
// 入栈操作:压入当前遍历元素对应的下标
stack = append(stack, i)
}
return res
}
时间复杂度:O(n),n 为数组的长度,正向遍历数组一遍,对于每个元素,最多有一次进栈和出栈的操作
空间复杂度:O(n)