夯实算法-每日温度

news2024/10/5 13:53:43

题目：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^51<= temperatures.length<=105
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;
}
复制代码