[프로그래머스] 더 맵게 - python

코드

시간초과 코드

from collections import deque
def solution(scoville, K):
    answer = 0
    scoville = deque(scoville)
    while True:
        scoville = deque(sorted(scoville))
        if scoville[0] >= K:
            break
        v1 = scoville.popleft()
        v2 = scoville.popleft()
        scoville.append(v1+(v2*2))
        answer += 1
    return answer

심지어 몇몇테스트 런타임에러도 발생

deque 사용해서 새로운 스코빌지수 들어올 때 마다 정렬해줌 

내가 봐도 구린 코드 .. 

 

 

시간초과 해결 코드

from heapq import heapify, heappush, heappop
def solution(scoville, K):
    answer = 0
    heapify(scoville)
    while True:
        if scoville[0] >= K:
            break
        v1 = heappop(scoville)
        v2 = heappop(scoville)
        heappush(scoville, v1+(v2*2))
        answer += 1
    return answer

최소힙을 이용해서 시간초과는 해결했지만 몇 개의 테스트에서 런타임에러 발생

 

해결

from heapq import heapify, heappush, heappop
def solution(scoville, K):
    answer = 0
    heapify(scoville)
    while True:
        if scoville[0] >= K:
            break
        if len(scoville) <= 1 and scoville[0] < K:
            return -1
        v1 = heappop(scoville)
        v2 = heappop(scoville)
        heappush(scoville, v1+(v2*2))
        answer += 1
    return answer

아앙 문제 제대로 안읽어서 그럼

모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다. 부분 적용해주니 문제 해결됨

 

참고자료

https://www.daleseo.com/python-heapq/