【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 ) O(N) O(N)

最坏情况下，每一天的温度比前一天更低，总共运行 N ∗ ( N − 1 ) / 2 N*(N-1)/2 N∗(N−1)/2 次 ： O ( N 2 ) O(N^2) O(N2)

这个方法在 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 ) O(N) O(N)