Leetcode739. Daily Temperature-栈解法

Leetcode739. Daily Temperatures

题目

题目链接

思路

1.暴力循环,对每个温度都向后搜索,找到第一个比它大的值
2.使用栈,从前到后,将温度压入栈中,遇到比它大的温度时弹出,记录大温度的索引,
这样既能保证是第一个比它大的,又防止了两重循环

复杂度

暴力循环的时间复杂度是$O(n^2)$
使用栈:
时间复杂度,从前到后遍历一次即完成操作,因此为$O(n)$
空间复杂度,新建了栈,最坏情况是所有元素都入栈,严格倒序的数组如[5,4,3,2,1],$O(n)$

代码

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        stack = []  # 栈
        result = [0 for _ in range(len(T))]  # 保存结果
        for idx, val in enumerate(T):
            # 把比它小的都弹出,再入栈
            while len(stack) and stack[-1][1] < val:
                k, v = stack.pop()
                result[k] = idx - k

            stack.append((idx, val))
        
        return result