[99클럽] 5일차 TIL

오늘 풀어볼 문제는 프로그래머스의 더 맵게입니다.

 

힙을 사용한 문제풀이를 진행하였습니다.

 

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

프로그래머스

 

 

 

import heapq

def solution(scoville, K):
    answer = 0
    
    heapq.heapify(scoville)
    while scoville:
        first = heapq.heappop(scoville)
        if first >= K:
            return answer
        second = heapq.heappop(scoville)
        score = first + (second * 2)
        heapq.heappush(scoville, score)
        answer += 1
        if len(scoville) == 1 and scoville[0] < K:
            return -1

 

힙을 사용하여 풀이하였습니다.