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

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

코딩테스트 연습 - 더 맵게

 

힙 배우고 호기롭게 풀기에 도전했으나 정확성 테스트는 다 맞았는데 효율성은 다 틀렸다🥲

효율성 그게 뭔데...!! 흑 ㅠㅠ 일단 정확성 다 맞은 코드... 좀 더 공부해서 효율성도 다 통과하겠어...🥲

 

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while True:
        # 모든 음식의 스코빌 지수가 K 이상인지 확인
        food = 0
        for i in range(len(scoville)):
            if scoville[i] >= K:
                food += 1
        if food == len(scoville): # 스코빌 지수 K 이상인 음식 개수와 스코빌의 개수가 동일하면
            break # 반복문 종료

        if len(scoville) <= 1: # 음식 개수가 1과 같거나 작아지면 불가능한 것이므로
            return -1 # -1 리턴

        temp1 = heapq.heappop(scoville) # 스코빌 지수 제일 작은 음식
        temp2 = heapq.heappop(scoville) # 두 번째로 작은 음식
        heapq.heappush(scoville, temp1+(temp2*2)) # 섞은 새로운 음식 push
        answer += 1 # answer 1 증가
    return answer