力扣热题100_栈_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]

解题思路

单调栈

• 首先，将答案数组 ans 全部赋值为 0。然后遍历数组每个位置元素。
• 如果栈为空，则将当前元素的下标入栈。
• 如果栈不为空，且当前数字大于栈顶元素对应数字，则栈顶元素出栈，并计算下标差。
• 此时当前元素就是栈顶元素的下一个更高值，将其下标差存入答案数组 ans 中保存起来，判断栈顶元素。
• 直到当前数字小于或等于栈顶元素，则停止出栈，将当前元素下标入栈。
• 最后输出答案数组 ans。

解题代码

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        n = len(T)
        stack = []
        ans = [0 for _ in range(n)]
        for i in range(n):
            while stack and T[i] > T[stack[-1]]:
                index = stack.pop()
                ans[index] = (i-index)
            stack.append(i)
        return ans

---

参考资料：datawhalechina