https://leetcode.com/problems/daily-temperatures/

給一個每天溫度的陣列,回傳一個陣列,值代表:今天開始「要過幾天」溫度才會比今天高

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]


December 29, 2023

Screenshot 2023-12-29 at 5.27.52 PM.png

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        stack = [] # monotonic decreasing stack, store index
        res = [0] * len(temperatures)

        for i, temp in enumerate(temperatures):
            while stack and (temperatures[stack[-1]] < temp):
                res[stack[-1]] = i - stack[-1]
                stack.pop()
            stack.append(i)

        return res

time O(n)

space O(n)