Daily Temperature

739. Daily Temperature

Difficulty: Medium

Related Topics: Array, Stack, Monotonic Stack

Given an array of integers temperatures represents the daily temperatures, return an array answer such that answer[i] is the number of days you have to wait after the ith day to get a warmer temperature. If there is no future day for which this is possible, keep answer[i] == 0 instead.

Example 1:

Input: temperatures = [73,74,75,71,69,72,76,73] Output: [1,1,4,2,1,1,0,0]

Example 2:

Input: temperatures = [30,40,50,60] Output: [1,1,1,0]

Example 3:

Input: temperatures = [30,60,90] Output: [1,1,0]

Code:

class Solution:
    def dailyTemperatures(self, temp: List[int]) -> List[int]:
        stack = []
        res = [ 0 for _ in range(len(temp))]

        for i in range(len(temp)):
            while stack and temp[i] > temp[stack[-1]]:
                ind = stack.pop()
                res[ind] = i-ind
            stack.append(i)
        
        return res