본문 바로가기

코딩/leetcode

[leetcode]22 일일 온도

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

 

Daily Temperatures - LeetCode

Can you solve this real interview question? Daily Temperatures - Given an array of integers temperatures represents the daily temperatures, return an array answer such that answer[i] is the number of days you have to wait after the ith day to get a warmer

leetcode.com

 

코드

class Solution(object):
    def dailyTemperatures(self, temperatures):
        result =[]
        stack  =[]
        
        for i in range(len(temperatures)):
            while stack and temperatures[stack[-1]]<temperatures[i]:
                idx = stack.pop()
                result_idx=i-idx
                result[idx]=result_idx
                
                
            stack.append(i)
            result.append(0)
            
        return result