[Leetcode] 739. Daily Temperatures

Leetcode 739. Daily Temperatures

Daily Temperatures - LeetCode

Solution

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        stack = []
        result = [0]*len(T)
        for i, n in enumerate(T):
            while stack and n > T[stack[-1]]:
                top = stack.pop()
                result[top] = i - top
            stack.append(i)
        return result

1. result 리스트를 T의 길이만큼 0으로 초기화
2. 스택이 비어있으면 append해주고, 비어있지 않으면서 고점이 오면 stack.pop()하여 인덱스의 차를 result에 저장해줌

Reference

파이썬 알고리즘 인터뷰

파이썬 알고리즘 인터뷰 국내도서 저자 : 박상길 출판 : 책만 2020.07.15 상세보기