题目链接:https://leetcode-cn.com/problems/iIQa4I/
请根据每日 气温 列表 temperatures ,重新生成一个列表,要求其对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
解题思路
- 定义温度函数
/**
* @param {number[]} 每日温度数组[73,74,75,71,69,72,76,73]
* @return {number[]} 每个温度等待天数列表[1,1,4,2,1,1,0,0]
*/
var dailyTemperatures = function (T) {
}
- 先创建一个单调栈 stack = [0],存储传入数组所有元素的索引
// 定义一个单调栈用于存储所传入数组的索引
let stack = [0]
let count = 1
- 创建结果数组,默认将结果数组用 0 填充
cont len = T.length
// 创建结果数组,长度和传入温度一致,初始值都填充为0
let arr = new Array(len).fill(0)
- 遍历传入的 数组 T,使用temp比较栈顶值:
- 通过while语句来循环比较:如果栈顶值小,计算日期差,并存储到结果数组,否则出栈重复操作
- 每次循环完将当前值温度索引值入栈即可
for(let i = 1; i < len; i++) {
let temp = T[i]
// 通过while循环比较
// 当count大于0,且temp > stack栈顶值时,进入循环并计算日期差,最后存入结果数组
// 否则重复
while(count && temp > T[stack[count-1]]) {
let index = stack.pop()
count--
arr[index] = i - index
}
// 处理完毕,当前值温度入栈(等待找到后续更大温度)
stack.push(i)
count++
}
return arr
说明:
- stack用于记录需要比较的温度索引,如果温度比较已经完成则出栈,只保留需要比较的温度