[프로그래머스/Programmers] 더 맵게 - Python

Link

코딩테스트 연습 - 더 맵게

 

Solution

해당 문제를 heapq를 사용하지 않고 해결해보려고 했으나 계속 효율성 문제에 걸려 결국 heapq를 사용하였다.

heapq에 대한 자세한 설명은 하단 첨부링크를 확인하기 바란다.

def solution(scoville, K):
    answer = 0

    scoville.sort()

    while True:
        if scoville[0] < K:
            scoville[1] = scoville[0] + scoville[1] * 2
            del scoville[0]
            answer += 1

        scoville.sort()

        t = scoville[0] >= K

        if t == False and len(scoville) == 1:
            return -1
        elif t == True:
            return answer

import heapq

def solution(scoville, K):
    answer = 0

    heapq.heapify(scoville)

    while True:
        if scoville[0] < K:
            t = heapq.heappop(scoville) + heapq.heappop(scoville) * 2
            heapq.heappush(scoville, t)
            answer += 1

        if len(scoville) == 1 and scoville[0] < K:
            return -1

        if scoville[0] >= K:
            return answer

🔑 key point: heapq

첨부

https://littlefoxdiary.tistory.com/3