Leetcode-739. Daily Temperatures

最新推荐文章于 2025-12-13 21:50:15 发布

原创最新推荐文章于 2025-12-13 21:50:15 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 专栏收录该内容

10 篇文章

订阅专栏

本文针对LeetCode上的一道题目——每日气温问题进行了解析，介绍了一种利用栈来求解的方法。具体步骤包括初始化一个栈来存储位置，逆序遍历温度数组，并通过比较栈顶元素与当前元素的大小来确定答案。

Original link: https://leetcode.com/problems/daily-temperatures/

Given a list of daily temperatures, return a list such that tell how many days you would have to wait until a warmer temperature.

Initialize a stack (list) to store position.
Iterate the array T in the reverse sequence, the stack will pop out position where the temperature (T[stack[-1]) is less than the current temperature (T[i]).
The remained last element in stack is the position where the temperature is larger than the current one. If the stack is empty, it means that elements after the current one are all smaller.

Code:

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