739. Daily Temperatures

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
    
        stack = []
        res = [0]*len(T)
        
        for i, temperature in enumerate(T):
        
            while stack and temperature > T[stack[-1]]:
                index = stack.pop()
                res[index] = i - index # write answer
                
            stack.append(i)     
            
        return res