[프로그래머스 Python] Lv 1. 과일 장수

---

문제 설명

과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 1점부터 k점까지의 점수로 분류하며, k점이 최상품의 사과이고 1점이 최하품의 사과입니다. 사과 한 상자의 가격은 다음과 같이 결정됩니다.

• 한 상자에 사과를 m개씩 담아 포장합니다.
• 상자에 담긴 사과 중 가장 낮은 점수가 p (1 ≤ p ≤ k)점인 경우, 사과 한 상자의 가격은 p * m 입니다.

과일 장수가 가능한 많은 사과를 팔았을 때, 얻을 수 있는 최대 이익을 계산하고자 합니다.(사과는 상자 단위로만 판매하며, 남는 사과는 버립니다)
예를 들어, k = 3, m = 4, 사과 7개의 점수가 [1, 2, 3, 1, 2, 3, 1]이라면, 다음과 같이 [2, 3, 2, 3]으로 구성된 사과 상자 1개를 만들어 판매하여 최대 이익을 얻을 수 있습니다.

• (최저 사과 점수) x (한 상자에 담긴 사과 개수) x (상자의 개수) = 2 x 4 x 1 = 8

사과의 최대 점수 k, 한 상자에 들어가는 사과의 수 m, 사과들의 점수 score가 주어졌을 때, 과일 장수가 얻을 수 있는 최대 이익을 return하는 solution 함수를 완성해주세요.

제한사항

• 3 ≤ k ≤ 9
• 3 ≤ m ≤ 10
• 7 ≤ score의 길이 ≤ 1,000,000
  • 1 ≤ score[i] ≤ k
• 이익이 발생하지 않는 경우에는 0을 return 해주세요.
• 문제 보러가기

문제 풀이

(1) 코드 작성

def solution(k, m, score):
    score = sorted([s for s in score if s <= k])[::-1]
    apple_box, box = list(), list()
    # 과일을 m개로 묶는 과정
    for s in score:
        box.append(s)
        if len(box) // m == 1:
            apple_box.append(box)
            box = []
    return sum([min(apples) for apples in apple_box]) * m

(2) 코드 리뷰

위 코드는 과일을 m 개로 묶는 과정(반복문)을 포함한다. 과일을 묶어서 판매할 때 가격은 최저 사과 점수를 기준으로 측정된다. 그리고 전체 사과의 개수가 모두 판매되지 못할 수 있다. 이 점을 고려하여 score을 내림차순으로 정렬하였고 m 개씩 묶어 m 개씩 묶인 각 리스트에서의 최저 과일 점수의 합과 과일 박스당 과일의 수를 곱해주었다. 이번 문제는 다른 문제들보다는 쉬운 편이었다.

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.