본문 바로가기

코딩 테스트/프로그래머스

[프로그래머스 Python] - 더 맵게(heap)

반응형

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

 

코딩테스트 연습 - 더 맵게

매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같

programmers.co.kr

import heapq

def solution(scoville, K):
    cnt = 0
    heapq.heapify(scoville)
    
    while(1):
        min1 = heapq.heappop(scoville)
        if(min1<K and len(scoville)>=1):
            min2 = heapq.heappop(scoville)
            heapq.heappush(scoville,min1+min2*2)
            cnt+=1
        else: break
    if(scoville[0]>=K): return cnt
    else: return -1

 

정렬보다 더 빠른 알고리즘으로 힙을 사용했다. 

반응형