739. Daily Temperatures

Given a list of daily temperatures T, return a list such that, for each day in the input, tells you how many days you would have to wait until a warmer temperature. If there is no future day for which this is possible, put 0 instead.

For example, given the list of temperatures T = [73, 74, 75, 71, 69, 72, 76, 73], your output should be [1, 1, 4, 2, 1, 1, 0, 0].

Note: The length of temperatures will be in the range [1, 30000]. Each temperature will be an integer in the range [30, 100].

 

class Solution(object):
    def dailyTemperatures(self, T):
        """
        :type T: List[int]
        :rtype: List[int]
        """
        
        l = [0] * len(T)
        
        s = []
        
        for i, t in enumerate(T):
            if len(s) == 0:
                s.append([i,t])
            else:
                while s and t > s[-1][1]:
                    top = s.pop()
                    l[top[0]] = i - top[0]
                s.append([i,t])
        
        return l