Haribo ML, AI, MATH, Algorithm

더 맵게

2021-01-18
Haribo

더 맵게

코드

import heapq
def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while scoville :
        first = heapq.heappop(scoville)
        if not scoville and first < K :
            return -1
        elif first >= K : 
            break
        second = heapq.heappop(scoville)
        new_food = first + second * 2
        heapq.heappush(scoville, new_food)
        answer += 1
    return answer

heapq

heapq 라이브러리는 이런 문제풀이에서 정말 좋은 알고리즘이라 꼭 공부 해서 외워야합니다.

heapq 공식 문서


Similar Posts

이전 포스트 다리를 지나는 트럭

다음 포스트 멀쩡한 사각형

Comments