[파이썬] 프로그래머스: 과일 장수

최대 이익값을 반환하기 위해 주어진 과일 점수를 높은순으로 정렬하고,

앞에서 m개씩 끊어가며 규칙에 따른 해당 박스의 이익을 answer에 누적한다.

단, 마지막 박스에 m개가 가득 차지 않을 경우 연산을 수행하지 않기 위해서

while문으로 중단 조건을 선제시해준다.

def solution(k,m,score):
	answer=0
    start, end = 0, m
    score.sort(reverse=True)
    
    while (len(score[start:end]) == m):
    	answer += (min(score[start:end])*m)
        start += m
        end += m
        
    return answer

 

프로그래머스