삽집하는 개발들/알고리즘

[42일차][프로그래머스][135808]과일 장수

악투 2023. 8. 10. 18:58
반응형
import math

def solution(k, m, score):
    sort_score = sorted(score, reverse=True)
    box_count = math.floor(len(score) / m)
    result = 0

    if box_count == 0:
        return 0
    else:
        for idx1 in range(box_count):
            if idx1 == 0:                
                result += min(sort_score[idx1:idx1+m]) * m
            else:
                result += min(sort_score[idx1*m:idx1*m+m]) * m
                      
    return result
반응형