Leetcode —【中等】739. 每日温度

 

根据每日 气温 列表，重新生成一个列表。对应位置的输出为：要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 0 来代替。

 

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

 

题目理解： 数组中元素与下一个比它大的元素之间的距离

 

复杂度解析：该方法只需要对数组进行一次遍历，每个元素最多被压入和弹出堆栈一次，所以算法复杂度为O(N)。

 

维护递减栈，后入栈的元素总比栈顶元素小。

• 比对当前元素与栈顶元素的大小
• 若当前元素 < 栈顶元素：入栈
• 若当前元素 > 栈顶元素：弹出栈顶元素，记录两者下标差值即为所求天数

这里用栈记录的是 T 的下标。

 

Python：

 

class Solution (object):

    def dailyTemperatures (self, T):

        """

        :type T: List[int]

        :rtype: List[int]

        """

        stack = list()

        t_length = len(T)

        res_list = [ 0 for _ in range(t_length)]

       

        for key, value in enumerate(T):    

            if stack:

                while stack and T[stack[ -1]] < value:

                    res_list[stack[ -1]] = key - stack[ -1]

                    stack.pop()

            stack.append(key)

        return res_list