栈学习力扣739. 每日温度

题目

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

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

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

来源：力扣（LeetCode）
链接: 题目链接
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

暴力解法1（会超时）

时间复杂度：O（n^2)
空间复杂度：O（n）

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

暴力解法2

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        n=len(T)
        res=[0]*n
        d={}
        for i in range(n-1,-1,-1):
            d[T[i]]=i
            tmp=[d[t]-i for t in range(T[i]+1,101) if t in d]
            res[i]=(min(tmp) if tmp else 0)
        return res

单调栈

时间复杂度：O（n）
空间复杂度：O（n）

算法流程

1，首先建立一个单调栈stack，在遍历字符串T时，stack用来记录第一个比i大的元素的下标。
2，初始化res=[0]*len(T)
3,遍历T时，只要stack不为空 and T[stack[-1]] < T[i]:，则res[stack[-1]] = i - stack[-1]，并且要将T[stack[-1]]元素的下标在stack中出栈。
4，如果T[stack[-1]] >= T[i] 或者stack为空时，对应i入栈。

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