739. 每日温度

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]

此题和496. 下一个更大元素 I非常类似。
思路也是用栈的方式，用普通的for循环会提示超时。
思路：

• 初始化一个值全为0的列表res和空栈stack 。
• 第一个温度值的索引号入栈， 第二个值和上一个值对比。
• 如果后一个值大于前一个值，则将小值的索引号弹出。
• 否则如果后一个值小于前一个值，则将后一个值直接入栈，直到找到一个大于栈底元素的值，将该大值的索引减去栈底元素的索引作为res中的弹出小值的索引号对应的值。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        res=[0]*len(temperatures)
        stack=[]
        for idx,item in enumerate(temperatures):
            while(stack and item>temperatures[stack[-1]]):
                index=stack.pop()
                res[index]=idx-index
            stack.append(idx)
        return res