Leetcode739. 每日温度

最新推荐文章于 2025-02-02 14:16:15 发布

会流泪de鱼

最新推荐文章于 2025-02-02 14:16:15 发布

阅读量747

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： java 算法数据结构

本文链接：https://blog.youkuaiyun.com/sunhaiting666/article/details/126706788

Leetcode 专栏收录该内容

214 篇文章

订阅专栏

该博客详细介绍了如何利用单调栈解决LeetCode 739每日温度问题，通过示例解释了算法思路，即维护一个存储下标的单调栈，从栈底到栈顶的下标对应温度依次递减，从而找出每个温度的下一个更高温度出现的天数。在遍历过程中，当遇到更高的温度时，更新等待天数并进栈，最后得到答案。提供的Java代码展示了具体实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Leetcode739. 每日温度

题目：
给定一个整数数组 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]

题解：
可以维护一个存储下标的单调栈，从栈底到栈顶的下标对应的温度列表中的温度依次递减。如果一个下标在单调栈里，则表示尚未找到下一次温度更高的下标。

正向遍历温度列表。对于温度列表中的每个元素 $t e m p er a t u res [i]$ ，如果栈为空，则直接将 $i$ 进栈，如果栈不为空，则比较栈顶元素 $p re v I n d e x$ 对应的温度 $t e m p er a t u res [p re v I n d e x]$ 和当前温度 $t e m p er a t u res [i]$ ，如果 $t e m p er a t u res [i] > t e m p er a t u res [p re v I n d e x]$ ，则将 $p re v I n d e x$ 移除，并将 $p re v I n d e x$ 对应的等待天数赋为 $i - p re v I n d e x$ ，重复上述操作直到栈为空或者栈顶元素对应的温度小于等于当前温度，然后将 $i$ 进栈。

为什么可以在弹栈的时候更新 $an s [p re v I n d e x]$ 呢？因为在这种情况下，即将进栈的 $i$ 对应的 $t e m p er a t u res [i]$ 一定是 $t e m p er a t u res [p re v I n d e x]$ 右边第一个比它大的元素，试想如果 $p re v I n d e x$ 和 $i$ 有比它大的元素，假设下标为 $j$ ，那么 $p re v I n d e x$ 一定会在下标 $j$ 的那一轮被弹掉。

由于单调栈满足从栈底到栈顶元素对应的温度递减，因此每次有元素进栈时，会将温度更低的元素全部移除，并更新出栈元素对应的等待天数，这样可以确保等待天数一定是最小的。

以下用一个具体的例子帮助读者理解单调栈。对于温度列表 $[73, 74, 75, 71, 69, 72, 76, 73]$ ，单调栈 $stack\textit{stack}$ 的初始状态为空，答案 $ans\textit{ans}$ 的初始状态是 $[0, 0, 0, 0, 0, 0, 0, 0]$ ，按照以下步骤更新单调栈和答案，其中单调栈内的元素都是下标，括号内的数字表示下标在温度列表中对应的温度。

当 $i = 0$ 时，单调栈为空，因此将 0 进栈。

$stack=[0(73)]\textit{stack}=[0(73)]$

$ans=[0,0,0,0,0,0,0,0]\textit{ans}=[0,0,0,0,0,0,0,0]$

当 $i = 1$ 时，由于 74 大于 73，因此移除栈顶元素 0，赋值 $an s [0] := 1 - 0$ ，将 1 进栈。

$stack=[1(74)]\textit{stack}=[1(74)]$

$ans=[1,0,0,0,0,0,0,0]\textit{ans}=[1,0,0,0,0,0,0,0]$

当 $i = 2$ 时，由于 75 大于 74，因此移除栈顶元素 1，赋值 $an s [1] := 2 - 1$ ，将 2 进栈。

$stack=[2(75)]\textit{stack}=[2(75)]$

$ans=[1,1,0,0,0,0,0,0]\textit{ans}=[1,1,0,0,0,0,0,0]$

当 $i = 3$ 时，由于 71 小于 75，因此将 3 进栈。

$stack=[2(75),3(71)]\textit{stack}=[2(75),3(71)]$

$ans=[1,1,0,0,0,0,0,0]\textit{ans}=[1,1,0,0,0,0,0,0]$

当 $i = 4$ 时，由于 69 小于 71，因此将 4 进栈。

$stack=[2(75),3(71),4(69)]\textit{stack}=[2(75),3(71),4(69)]$

$ans=[1,1,0,0,0,0,0,0]\textit{ans}=[1,1,0,0,0,0,0,0]$

当 $i = 5$ 时，由于 72 大于 69 和 71，因此依次移除栈顶元素 4 和 3，赋值 $an s [4] := 5 - 4$ 和 $an s [3] := 5 - 3$ ，将 5 进栈。

$stack=[2(75),5(72)]\textit{stack}=[2(75),5(72)]$

$ans=[1,1,0,2,1,0,0,0]\textit{ans}=[1,1,0,2,1,0,0,0]$

当 $i = 6$ 时，由于 76 大于 72 和 75，因此依次移除栈顶元素 5 和 2，赋值 $an s [5] := 6 - 5$ 和 $an s [2] := 6 - 2$ ，将 6 进栈。

$stack=[6(76)]\textit{stack}=[6(76)]$

$ans=[1,1,4,2,1,1,0,0]\textit{ans}=[1,1,4,2,1,1,0,0]$

当 $i = 7$ 时，由于 73 小于 76，因此将 7 进栈。

$stack=[6(76),7(73)]\textit{stack}=[6(76),7(73)]$

$ans=[1,1,4,2,1,1,0,0]\textit{ans}=[1,1,4,2,1,1,0,0]$

java代码：

  /**
     * @param temperatures
     * @return
     */
    public static int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] ans = new int[len];
        Deque<Integer> stack = new LinkedList<>();
        for (int i = 0; i < len; i++) {
            while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
                Integer pop = stack.pop();
                ans[pop] = i - pop;
            }
            stack.push(i);
        }
        return ans;
    }