프로그래머스 연습 문제 - 더 맵게

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

코딩테스트 연습 - 더 맵게

heap 문제

 

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while scoville:
        scov1 = heapq.heappop(scoville)
        

        if scov1 > K:
            break
        if len(scoville) == 0:
            return -1
        scov2 = heapq.heappop(scoville)
        heapq.heappush(scoville, scov1+scov2*2)
        answer += 1
        
    return answer

 

heap에 대해서 공부할 수 있었다.

 

 

#heap

우선순위 큐 알고리즘. 최솟값, 최댓값을 호출해야 하면 굉장히 유용하다.

heapq.heapify(scoville)

기존의 리스트를 heap으로 바꿔준다.

 

heapq.heappop(socville)

socville에서 삭제하는 함수. 최소값을 삭제한다.

 

heapq.heappush(scoville)

socville에서 원소를 추가하는 함수.

 

 

자세한건 따로 포스팅!