Thomas Ngo

Software Engineer

less than 1 minute read

Problem Statement

problem

My note note

Intuition

My initial thought is to use a stack to keep track of indices while traversing the temperatures array.

Approach

I will iterate through the temperatures array and use a stack to store the indices of the temperatures for which we haven’t found a warmer day yet. For each temperature, I’ll compare it with the temperatures at the indices stored in the stack. If the current temperature is higher, I’ll update the result for the corresponding index and pop that index from the stack. This process ensures that we find the nearest warmer day for each temperature.

Complexity

  • Time complexity: O(n), where n is the length of the temperatures array. We iterate through the array once.

  • Space complexity: O(n), in the worst case, all temperatures could be stored in the stack.

Code

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        stack = []
        res = [0] * len(temperatures)
        for i, temp in enumerate(temperatures):
            while stack and temperatures[stack[-1]] < temp:
                j = stack.pop()
                res[j] = i - j
                
            stack.append(i)
        return res

You may also enjoy