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]

法一、暴力

• 时间复杂度 O(n^2)

var dailyTemperatures = function(temperatures) {
    let res = new Array(temperatures.length).fill(0); // 初始化结果数组
    for (let i = 0; i < temperatures.length; i++) {
        for (let j = i + 1; j < temperatures.length; j++) {
            if (temperatures[j] > temperatures[i]) {
                res[i] = j - i; // 记录当前温度的天数差
                break; // 找到更温暖的温度后跳出内层循环
            }
        }
    }
    return res; // 返回结果数组
};

法二、单调栈

• 时间复杂度 O(n)，空间复杂度O(n)

const dailyTemperatures = (T) => {
  // 初始化结果数组，长度与输入数组相同，初始值为0
  const res = new Array(T.length).fill(0);
  const stack = [];// 初始化一个栈，用于存储温度的索引
  // 从后往前遍历温度数组
  for (let i = T.length - 1; i >= 0; i--) {
    // 当栈不为空且当前温度大于等于栈顶索引对应的温度时，弹出栈顶元素
    while (stack.length && T[i] >= T[stack[stack.length - 1]]) {
      stack.pop();
    }
    // 如果栈不为空，说明找到了一个更温暖的温度
    if (stack.length) {
      res[i] = stack[stack.length - 1] - i;// 计算当前温度到下一个更温暖温度的天数
    }
    stack.push(i); // 将当前索引推入栈中
  }
  return res;
}