【LeetCode 算法笔记】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]

示例 3:
输入: temperatures = [30,60,90]
输出: [1,1,0]

提示：
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100

暴力解法

对于每一个温度值，都向后进行扫描，直到找到比这个值更大的值。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        res = []
        n = len(temperatures)
        for i in range(n):
            j = 0
            finded = False
            while(i+j < n):
                if temperatures[i] < temperatures[i+j]:
                    finded = True
                    break
                else:
                    j += 1
            res.append(j if finded else 0)
        return res

算法复杂度：
最好的情况下，每一天的温度比前一天更高：$O(N)$
最坏情况下，每一天的温度比前一天更低，总共运行 $N\ast (N-1)/2$ 次 ：$O(N^2)$

这个方法在 LeetCode 中提交的话，会超时无法通过！

栈

先把 下标存到栈里，如果栈顶元素遇到了更高的温度，持续抛出

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        stack = []
        n = len(temperatures)
        res = [0]*n
        for i in range(n):
            while(stack and temperatures[stack[-1]] < temperatures[i]):
                k = stack.pop()
                res[k] = (i-k)
            stack.append(i)
        return res

算法复杂度：$O(N)$