더 맵게

섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)

코딩테스트 연습 - 더 맵게 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

나의 코드

def solution(scoville, k):

    answer = 0

    scoville.sort()

    while scoville[0]<k:

        answer+=1

        new_scoville= scoville.pop(0)+(scoville.pop(0)*2)

        scoville.append(new_scoville)

        scoville.sort()

    #print(scoville)

    print(answer)

    return answer

solution([1, 2, 3, 9, 10, 12], 7)

 

수정한 코드

import heapq

def solution(scoville, k):

    answer = 0

    #일다 스코빌지수 오름차순으로 정렬부터

    scoville.sort()

    while scoville[0] < k:

        #스코빌지수를 k이상으로 만들 수 없는 경우

        if len(scoville) == 1:

            return -1

       

        else:

            answer+=1

            heapq.heappush(scoville, heapq.heappop(scoville)+(heapq.heappop(scoville)*2))

    #print(scoville)

    return answer

solution([1,2,3,9,10,12], 7)

성공!!!!