LeetCode - 739. Daily Temperatures

739. Daily Temperatures

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        stack = [0]
        n = len(T)
        res = [0]*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