[9/3] 더 맵게

이경준·2021년 9월 3일
0

코테

목록 보기
93/140
post-custom-banner

레벨2 문제

내 코드

import heapq

def solution(sco, K):
    arr = []
    answer = 0
    
    for i in sco:
        heapq.heappush(arr, i)
    
    while ( arr[0] < K ):
        
        if ( len(arr) == 1 ):
            answer = -1
            break
        
        one = heapq.heappop(arr)
        two = heapq.heappop(arr)
        
        answer += 1
        num = one + (2 * two)
        heapq.heappush(arr, num)
    
    return answer

로직

  • 최소 힙 사용

피드백

  • min(arr) 보다 arr[0] 사용하는 것이 시간복잡도가 더 작다.
profile
The Show Must Go On
post-custom-banner

0개의 댓글