leetcode739 每日温度

题目

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。

示例

输入: temperatures = [73,74,75,71,69,72,76,73]

输出: [1,1,4,2,1,1,0,0]

解析

这道题的测试用例其实可以用这个:

temp := []int{1, 4, 3, 5, 5, 2, 3, 6}

这道题可以有两种方法,首先看从左到右的方法:

go 复制代码
func dailyFromLeft(temperature []int) []int {
	n := len(temperature)
	// 两个额外数组,一个存结果ans,这个默认已经初始化为0了,后面更新哪个位置,就将对应的下标间隔更新进去
	ans := make([]int, n)
	// 另一个就是栈,栈中存的是下标
	stack := []int{}
	for i, value := range temperature {
		for len(stack) > 0 && value > temperature[stack[len(stack)-1]] {
			j := stack[len(stack)-1]     // 之前存在栈里的下标
			stack = stack[:len(stack)-1] // 弹出栈顶
			ans[j] = i - j
		}
		stack = append(stack, i)
	}
	return ans
}

这道题愣是去理解的话,还是有点费劲,建议根据上面的那个测试用例来进行debug,首先在从左到右的顺序遍历过程中,栈中存的下标用j来表示,就相当于遍历到i的时候(此时j小i大,那距离就是i-j),就相当于在第一次遍历到的某个位置的时候,此时还不知道后面谁比他大的,遍历到第一个比它大的时候才进行计算,添加到ans中。

然后是从右到左的方法:

go 复制代码
func dailyTemperatures(temperatures []int) []int {
	n := len(temperature)
	// 两个额外数组,一个存结果ans,这个默认已经初始化为0了,后面更新哪个位置,就将对应的下标间隔更新进去
	ans := make([]int, n)
	// 另一个就是栈,栈中存的是下标
	stack := []int{}
	for i := n - 1; i >= 0; i-- {
		value := temperature[i]
		for len(stack) > 0 && value >= temperature[stack[len(stack)-1]] { // 从右到左遍历,当前的高度大于栈中的数据,就弹出栈中的小的数据
			stack = stack[:len(stack)-1]
		}
		if len(stack) > 0 {
			ans[i] = stack[len(stack)-1] - i // 将栈中存的比当前位置大的下标与当前位置下标相减
		}
		stack = append(stack, i) // stack中存的是下标
	}
	return ans
}
相关推荐
程序员Xu1 小时前
【LeetCode热题100道笔记】二叉树的右视图
笔记·算法·leetcode
笑脸惹桃花2 小时前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
阿维的博客日记2 小时前
LeetCode 48 - 旋转图像算法详解(全网最优雅的Java算法
算法·leetcode
GEO_YScsn2 小时前
Rust 的生命周期与借用检查:安全性深度保障的基石
网络·算法
程序员Xu2 小时前
【LeetCode热题100道笔记】二叉搜索树中第 K 小的元素
笔记·算法·leetcode
THMAIL3 小时前
机器学习从入门到精通 - 数据预处理实战秘籍:清洗、转换与特征工程入门
人工智能·python·算法·机器学习·数据挖掘·逻辑回归
Kevinhbr4 小时前
CSP-J/S IS COMING
数据结构·c++·算法
蕓晨4 小时前
set的插入和pair的用法
c++·算法
THMAIL4 小时前
深度学习从入门到精通 - AutoML与神经网络搜索(NAS):自动化模型设计未来
人工智能·python·深度学习·神经网络·算法·机器学习·逻辑回归
金古圣人4 小时前
hot100 滑动窗口
数据结构·c++·算法·leetcode·哈希算法