[Leetcode] - 739

Jisung Park·2021년 5월 10일
0
  • 크기 비교가 필요한 문제는 stack으로 풀 수 있다
  • stack으로 값을 언제 비교하고, 언제 push, pop 할지 고민하면 될듯

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        
        ans = [0 for _ in range(len(T))]
        stack = []
        for idx, t in enumerate(T):
            while stack and (t > stack[-1][1]):            
                vidx, vt = stack.pop()
                ans[vidx] = idx - vidx
                
            stack.append((idx, t))
            
                
        return ans

0개의 댓글