leetcode 739

문제

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

풀이

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        arr = []
        answer = [0] * len(temperatures)
        
        for i in range(len(temperatures)):
            while(arr and temperatures[i] > temperatures[arr[-1]]):
                last = arr[-1]
                answer[last] = i-last
                arr.pop()
                    
            arr.append(i)
            
        return answer

인덱스를 스택에 담아서 저장하는 방식으로 풀이했다. 개인적으로 굉장히 어렵다고 느꼈다.

다음 인덱스를 읽으며 스택의 맨 마지막값부터 차례로 비교한다.
스택에는 온도가 높은 인덱스부터 차례로 낮아지는 방향으로 인덱스가 쌓이기 때문에, 가장 뒷 값 부터 비교하며 while문으로 조건을 만족하지 않을 때 까지 비교해주었다.