[Python] 프로그래머스, 과일 장수 Lv.1 (feat.sort, min, 파이썬, 한줄코드)

반짝반짝 알고리즘, 인프준

 

 

 

 

과일 장수

문제 링크

프로그래머스

 

 

문제 설명

과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 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 해주세요.

 

 

입출력 예

k	m	score	result
3	4	[ 1, 2, 3, 1, 2, 3, 1 ]	8
4	3	[ 4, 1, 2, 2, 4, 4, 4, 4, 1, 2, 4, 2 ]	33

 

 

입출력 예 설명

입출력 예#1

문제의 예시와 같습니다.

 

입출력 예#2

다음과 같이 사과 상자를 포장하여 모두 팔면 최대 이익을 낼 수 있습니다.

사과상자	가격
[ 1, 1, 2 ]	1 x 3 = 3
[ 2, 2, 2 ]	2 x 3 = 6
[ 4, 4, 4 ]	4 x 3 = 12
[ 4, 4, 4 ]	4 x 3 = 12

따라서 ( 1 x 3 x 1 ) + ( 2 x 3 x 1 ) + ( 4 x 3 x 2 ) = 33을 return 합니다.

 

 

 

 

해결 과정

1. 최대 이익 값을 반환하는 문제입니다. 결과 값을 answer로 선언해 주고 score list를 m 개씩 담아주기 위해 start, end를 선언해 주었습니다. 그리고 score list를 점수가 높은 순으로 정렬해 주었습니다.

 

2. 사과의 수가 m개를 만족할 때까지 while loop을 돌립니다. 담은 사과의 최솟값 min과 한 상자의 들어가는 사과의 수 m을 곱하여 결과 값에 누적시킵니다. 그리고 다음 상자를 담기 위해 start와 end에 m을 더해줍니다.

 

3. 사과의 수가 m개보다 적게 남았을 때 while문은 종료되고 결과 값을 반환합니다. while문 조건이 만족하지 않을 때는 선언했던 answer의 값 0이 그대로 반환됩니다.

 

def solution(k, m, score):
    answer = 0                  				# 최대 이익 값 (결과 값)
    start, end = 0, m           				# 사과의 수 시작, 끝
    score.sort(reverse=True)    				# 사과 점수 기준, 내림차순 정렬
    
    while( len(score[start:end]) == m ):        # 사과의 수가 만족할 때 loop
        answer += ( min(score[start:end]) * m ) # 최저점수 * 사과의 수, 누적
        start += m                              # 시작 + 사과의 수
        end += m                                # 끝 + 사과의 수

    return answer                               # 결과값 반환, 이익이 없으면 0

 

 

 

 

 

또 다른 풀이

 

Sum과 나머지 연산자 활용

def solution(k, m, score):
    return sum(sorted(score)[ len(score) % m::m ]) * m

 

Lambda 함수 활용

solution = lambda _, m, s: sum(sorted(s)[-m::-m]) * m

 

 

 

 

 

 

---

 

 

 

 

 

 

출처: 프로그래머스 코딩 테스트 연습,https://school.programmers.co.kr/learn/challenges