프로그래머스 파이썬 더 맵게

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

코딩테스트 연습 - 더 맵게

 

힙큐의 특성을 알아야 쉽게 해결이 가능하다

 

처음에 deque를 사용해서 미쳐버린 sort 50만번 작업을 할려했는데

다행히????? deque는 sort가 안된다고 하네요?????????????

 

heapq.heappush는 최소값을 우선해서 앞으로 넣기 때문에 이런 문제에 쓰기 딱 좋다.

 

heapqpop으로 야무지게 최소값 두개를 빼내서 a+(b*2) 를 하고

index에러가 나온다면 그냥 글러먹은것이니 -1를 출력

 

import heapq

def solution(scoville, K):
    answer = 0

    scoville.sort()

    heap = []

    for i in scoville:
        heapq.heappush(heap,i)



    while heap[0] < K:
        if heap[0] < K:
            try:
                a = heapq.heappop(heap)
                b = heapq.heappop(heap)

                heapq.heappush(heap,a+(b*2))
                answer += 1
            except IndexError:
                return -1


    return answer