[PYTHON] 프로그래머스 #42626 더 맵게

문제

코딩테스트 연습 - 더 맵게

 

나의 풀이

def solution(scoville, K):
    count = 0
    short = []

    for each in scoville:
        if each < K:
            short.append(each)

    short.sort()

    while True:
        if len(short) == 0:
            return -1
        count += 1
        if len(short) == 1:
            break
        else:
            addition = short[0] + short[1]
            short.pop()
            short.pop()
            if addition < K:
                insertsort_right(short, addition)

    return count

def insertsort_right(arr, a):
    for i in range(len(arr)):
        if arr[len(arr) - 1 - i] >= a:
            arr.insert(i, a)
            break

 

 

이렇게 편한게 있는지 몰랐다

heapify를 사용하면 자동으로 minheap구조로 바꿔준다

import heapq

def solution(scoville, K):
    #  scoville을 minheap구조로 바꿈
    heapq.heapify(scoville)
    count = 0

    while len(scoville) > 1:
        count += 1
        
        f = heapq.heappop(scoville)
        s = heapq.heappop(scoville)
        
        heapq.heappush(scoville, f+s*2)
        if scoville[0] >= K:
            return count

    return -1