leetcode-739. 每日温度

题目要求：

给定一个整数数组 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

链接：leetcode.cn/problems/da…

解题思路

第一种，想使用双指针，通过暴力解，结果超时了。

代码：

func dailyTemperatures(temperatures []int) []int {
	n := len(temperatures)
	if n <= 1 {
		return []int{0}
	}
	// 双指针，左侧指针挨个过，右侧指针右移
	l, r := 0, 0
	for r < n {
		cur := temperatures[l]
		for r < n && cur >= temperatures[r] {
			r++
		}
		if r == n && temperatures[r-1] <= cur {
			r = l
		}
		temperatures[l] = r - l
		l++
		r = l
	}
	return temperatures
}

看了题解，使用单调栈的方法，代码如下：

func dailyTemperatures(temperatures []int) []int {
	n := len(temperatures)
	if n <= 1 {
		return []int{0}
	}
	// 使用单调栈，按照逆序存储值的索引，每次从栈顶取值进行比较
	stack := []int{}
        //需要使用新的切片，不然temperatures最后比当前值小的没法表示
	ans := make([]int, n)
	for i := 0; i < n; i++ {
		for len(stack) > 0 && temperatures[i] > temperatures[stack[len(stack)-1]] {
			prevIndex := stack[len(stack)-1]
			stack = stack[:len(stack)-1]
			ans[prevIndex] = i - prevIndex
		}
		stack = append(stack, i)
	}

	return ans
}