[파이썬 알고리즘 인터뷰] 22. 일일 온도

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 코드 출처 : 파이썬 알고리즘 인터뷰 깃허브

 

---

22. 일일 온도

리트코드 739. Daily Temperatures 문제입니다.

leetcode.com/problems/daily-temperatures/

 

 

풀이

 stack에 인덱스를 저장하고 현재 인덱스에 해당하는 온도가 stack[-1] 인덱스에 해당하는 온도보다 크면 stack에서 값을 꺼내서 인덱스 차이를 계산합니다.

 

def dailyTemperatures(self, T):
    answer = [0] * len(T)
    stack = []
    for i, t in T:
        while stack and T[stack[-1]] < t:
            last = stack.pop()
            answer[last] = i - last
        stack.append(i)
    return answer