[프로그래머스 연습] 더 맵게 (Python)

https://programmers.co.kr/learn/courses/30/lessons/42626

코딩테스트 연습 - 더 맵게

 

import heapq
 
def solution(scoville, K):
    answer = 0
    # 효율성 테스트 실패
    # while min(scoville) < K:
    #     scoville.sort()
    #     scoville.append(scoville.pop(0) + (scoville.pop(0) * 2))
    #     answer += 1
    
    # heapq 모듈을 사용하면 python의 일반 리스트를 heap 구조로 사용할 수 있게해줌
    # 원소들이 항상 정렬된 상태로 추가되고 삭제됨
    # 가지고 있는 요소를 push, pop 할때마다 자동으로 정렬
    heapq.heapify(scoville)
    while scoville[0] < K:
        if len(scoville) > 1: # 예외처리 안해주면 정확성 감점
            heapq.heappush(scoville, heapq.heappop(scoville) + (heapq.heappop(scoville) * 2))
            answer += 1
        else:
            return -1
        
    return answer