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

[문제]

 

[잘못된 풀이 (런타임 에러)]

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    while scoville[0] < K:
        new_food = heapq.heappop(scoville) + heapq.heappop(scoville)*2
        heapq.heappush(scoville, new_food)
        answer+=1
    
    return answer

 

 

에러가 발생했을 때는 특이한 케이스를 생각해보면 된다. 문제의 제한사항에는 다음의 조건이 쓰여져있다.

• scoville의 길이는 2 이상 1,000,000 이하입니다.
• K는 0 이상 1,000,000,000 이하입니다.
• scoville의 원소는 각각 0 이상 1,000,000 이하입니다.
• 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다.

 

생각해볼 수 있는 예외 케이스

-> scoville의 길이가 2라면? -> scoville의 길이가 매우 길 때, 값이 여러번 중복되어 들어가있다면?     보통 LUT를 만들어 연산 및 수행시간을 줄임, 이 문제에서는 연산이 간단하므로 LUT 생성 X -> K가 0 이라면?     while문에 들어가지 않고 answer 값을 0으로 즉시 반환하기   -> K가 매우 큰 값이라서 연산을 아무리 해도 K의 값에 도달하지 못한다면?     -1을 반환하도록 코드 추가

 

논리를 짜는 것은 쉬울 수 있지만, 이런 제약사항을 반영하는 것이 코딩테스트의 변별력인 것 같다. 

 

 

[풀이]

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    # K 값이 0인 경우
    if K == 0:
        return 0
    
    while len(scoville) > 1 and scoville[0] < K:
        new_food = heapq.heappop(scoville) + heapq.heappop(scoville)*2
        heapq.heappush(scoville, new_food)
        answer+=1
    
    # 연산이 끝난 후에도 조건이 만족되지 않는 경우
    if scoville[0] < K:
        return -1
    
    return answer