夯实算法-每日温度

·  阅读 1101
夯实算法-每日温度

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第9天,点击查看活动详情

题目:LeetCode

给定一个整数数组 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]
复制代码

提示:

  • 1<= temperatures.length<=1051 <= temperatures.length <= 10^5
  • 30 <= temperatures[i] <= 100

解题思路

根据题意就是对于某一元素[i],要在[i+1, n-1]里面找j,使[j] > [i],要找第一个[j],如果没有就是0。

要提升效率,基本上遍历一次,将已遍历过的元素需要放入一个容器中,每次遍历时将当前元素与容器中的元素。题目要求是找最少的天数,因此应该与最近的元素进行比较,也即“后入先出”,可以用栈。

先将当前元素入栈,当遍历到后面的元素时,与栈顶的元素比较,如果元素大于栈顶元素,说明栈顶元素找到了结果,这个元素与栈顶之间的距离就是栈顶元素需要的天数。栈中的数据是数组下标。

栈中保存的是还没有找到结果的元素,也即较高的温度,遍历到当前元素[i]时,先与栈顶下标比较,如果[i]>[highTemps.peek()],那么栈顶这天就找到了结果,可以出栈了,要继续比较,直到栈里温度都大于[i]或者栈空了为止。

因为当前元素[i],肯定找不到结果,所以当前i是肯定要入栈的。

几个要点:

  • 栈里数据是下标,以方便计算天数,天数其实就是数组下标之差
  • 栈里保存的元素是较高的温度,也就是还没有找到结果的某几天
  • 当前元素肯定要入栈的
  • 当前元素与栈中元素进行比较,以找到栈中元素的结果(需要的天数)。

代码实现

public int[] dailyTemperatures(int[] temperatures) {
    int[] result = new int[temperatures.length];
    Stack < Integer > highTemps = new Stack <>();
    for (int i = 0; i < temperatures.length; i++) {
        while (!highTemps.isEmpty() && temperatures[i] > temperatures[highTemps.peek()]) {
            int prev = highTemps.pop();
            result[prev] = i - prev;
        }
        highTemps.push(i);
    }

    return result;
}
复制代码

复杂度分析

  • 空间复杂度:O(1)O(1)
  • 时间复杂度:O(n)O(n)

掘金(JUEJIN)  一起分享知识, Keep Learning!

收藏成功!
已添加到「」, 点击更改