Coding - Algo/python

[프로그래머스] 더 맵게 (python 파이썬)

jainn 2021. 6. 10. 00:55
728x90

문제

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

 

코딩테스트 연습 - 더 맵게

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

programmers.co.kr

 

풀이 및 소스코드

heapq 문제이다.

scoville은 리스트기 때문에 heapq.heapify를 사용하여 힙큐로 만들어준다.

import heapq
def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while 1:
        a = heapq.heappop(scoville)
        if a>=K:
            break
        if scoville:
            b = heapq.heappop(scoville)
            c = a+b*2
            heapq.heappush(scoville, c)
            answer+=1
        else:
            answer = -1
            break
    return answer
반응형