leetcode_739 每日温度

要求

请根据每日 气温 列表 temperatures ，请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高，请在该位置用 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]

核心代码

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

重点问题

解题思路：

我们使用s栈来记录当温度没有回升时候的温度的下标，当我们的遇到比栈中元素更大的温度的时候，弹出下标和现在的下标做运算，就能得到回暖时间差，比较经典的一道题。