力扣算法题解析:每日温度

142 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。 ​

题目链接:力扣

        请根据每日 气温 列表 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用于记录需要比较的温度索引,如果温度比较已经完成则出栈,只保留需要比较的温度