739. 每日温度

Difficulty
Medium
Tags
单调栈
URL
https://leetcode-cn.com/problems/daily-temperatures/
Star
请根据每日 气温 列表 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]
提示：
  • 1 <= temperatures.length <= 105
  • 30 <= temperatures[i] <= 100
通过次数206,851提交次数304,677

单调栈

思路
将下标存放到栈中，这样就可以计算相隔多少天了。
题解
class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        res = [0 for _ in temperatures]
        stack = []
        for i in range(len(temperatures)-1, -1, -1):
            while stack and temperatures[stack[-1]] <=temperatures[i]:
                stack.pop()
            res[i] = stack[-1] - i if stack else 0
            stack.append(i)

        return res