leetcode 739.每日温度

管科研一新生首刷leetcode，本题思路来源leetcode官解，写此帖初衷是为了帮助自己更好的理解和“背诵”代码，方便后期复习，希望各位大佬多多指教！

---

题目描述

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

最近在学习栈与深度优先搜索，题目出现在栈知识点的推荐题目中，虽然知道本题要使用栈来求解，但是想了半天还是不知道怎么用栈实现，故打算先使用暴力求解，但是最简单粗暴的方式出现了超出时间限制的情况。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        n = len(temperatures)
        answers = [0] * n 
        for i in range(n-1):
            for j in range(i+1,n):
                if temperatures[i]<temperatures[j]:
                    answers[i] = j-i
                    break
        return answers

能力所限，只能查看leetcode的题解。

解题思路

创建一个与temperatures等长的全零数组answer用来存放答案，创建一个空栈辅助本题操作。对temperatures数组进行遍历，当栈不为空，且当前遍历元素大于栈顶元素，则取出栈顶元素(temp)，answer数组temp索引位置的值赋为当前遍历元素所在索引与栈顶元素（temp）的差值；否则，将当前遍历元素的索引加入栈，直到上述循环结束。

代码实现

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

---

当时看了答案过了好久复习才打算梳理一下思路，忘记参考的到底是哪篇文章了，貌似是这个。https://leetcode.cn/problems/daily-temperatures/solution/739mei-ri-wen-du-python3-by-leetcode-wei-dzpu/