[LeetCode,Python] 739. 일일 온도(Daily Temperatures)

매일의 화씨온도 리스트 T를 입력받아서,

더 따뜻한 날씨를 위해서는 며칠을 더 기다려야 하는지를 출력하라.

 

Input: temperatures = [73,74,75,71,69,72,76,73]

Output: [1,1,4,2,1,1,0,0]

 

 

 

 

** 깔끔한 답안 **

 

 

 

1. 스택 값 비교

 

 

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        answer = [0] * len(temperatures)
        stack = []
        for i, cur in enumerate(temperatures):
            #현재 온도가 스택 값보다 높다면 정답 처리
            while stack and cur > temperatures[stack[-1]]:
                last = stack.pop()
                answer[last] = i - last
            stack.append(i)
        
        return answer

 

 

해설:

 

핵심은 현재의 온도가 전의 온도보다 높으면

answer에 값을 저장하고 stack에서 전의 온도 인덱스를 pop 하는 것이다.

 

이해가 어렵다면 인덱스와 온도 간의 그래프를 그려보는 것도 좋다.