더 맵게 - 프로그래머스 - 파이썬 풀이

문제 풀이

제일 안 매운 두 음식을 뽑아서 섞습니다.

이 프로세스를 모든 음식이 일정 기준(K)을 넘을 때까지 반복합니다.

만약 위의 방법이 실패하면 -1이 답입니다.

 

제일 안 매운 음식을 반복적으로 추출해야 하기 때문에

힙 구조를 이용해야 합니다.

학교에서는 주로 max-heap 을 배웠을텐데

여기서는 min 값을 지속적으로 이용해야 하기 때문에 min-heap을 이용합니다.

파이썬의 heapq 모듈은 기본적으로 min-heap입니다.

사용법은 help 키워드나 여기 링크를 참조하세요!

클린 코드 작성법

이번 문제는 클린 코드를 작성하기가 힘들었습니다.

실력자분이 나타나서 한 수 가르쳐줬으면 합니다.

1. check_conditions을 통해 break를 하나만 씁니다.
   1. 탈출 조건이 하나여야 while문을 이해하기 쉽습니다.
   2. 조건문 판별을 위한 변수명으로 is + 동사나 has + 명사를 이용하면 좋습니다.
2. 함수의 parameter은 2개 이하여야 합니다.
   1. OOP에서는 자신을 제외해서 1개 이하의 parameter을 가져야 합니다.
   2. 이 부분을 해결 못했습니다. 별도의 wrapper을 마련해 복잡성을 줄여야 합니다.

해답 코드

import heapq

def solution(scoville, K):
    heapq.heapify(scoville)
    mix_count = 0
    while True:
        food_one = heapq.heappop(scoville)
        is_finished, mix_count = check_conditions(food_one, scoville, K, mix_count)
        if is_finished:
            break
        mix_food(food_one, scoville)
        mix_count += 1
    return mix_count

def check_conditions(food_one, scoville, K, mix_count):
    is_finished = False
    if K <= food_one:
        is_finished = True
    has_no_food = not scoville
    if has_no_food and not is_finished:
        is_finished = True
        mix_count = -1
    return is_finished, mix_count

def mix_food(food_one, scoville):
    food_two = heapq.heappop(scoville)
    mixed_food = food_one + 2*food_two
    heapq.heappush(scoville, mixed_food)

도움이 됐다면 공감버튼을 눌러주세요! 질문이 있다면 댓글 달아주세요!

 

문제 링크

코딩테스트 연습 - 더 맵게 | 프로그래머스