739. Daily Temperatures

Given a list of daily temperatures, produce a list 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 temperatures = [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:
    def dailyTemperatures(self, temperatures):
        """
        :type temperatures: List[int]
        :rtype: List[int]
        """
        index, larger = [-1]*len(temperatures), [-1]*len(temperatures)
        larger[-1] = temperatures[-1]
        for i in range(len(temperatures)-2,-1,-1):
            for j in range(i+1, len(temperatures)):
                if temperatures[j]>temperatures[i]:
                    index[i] = j
                    larger[i] = temperatures[j]
                    break
                elif larger[j]>temperatures[i]:
                    index[i] = index[j]
                    larger[i] = larger[j]
                    break
                else:
                    j = index[j]
                    if j==-1:  break
                
        ret = [0]*len(temperatures)
        for idx, i in enumerate(index):
            if i!=-1:  ret[idx]=i-idx
        return ret
        
# s = Solution()
# print(s.dailyTemperatures([73, 74, 75, 71, 69, 72, 76, 73]))