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

⭐ 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/135808

프로그래머스

 

⭐ 풀이 코드

 

❌ 시간초과

def solution(k, m, score):
    answer = 0
    
    score.sort(reverse=True)
    
    for i in range(len(score)):
        temp = 0
        if (i+1)%m == 0:
            box = score[temp:i+1]
            temp = i+1
            answer += (min(box)*m)
    
    return answer

 

⭕ 통과

def solution(k, m, score):
    answer = 0

    score.sort(reverse=True) 

    for i in range(len(score)):
        if (i+1)%m == 1:
            box = []
        box.append(score[i])
    
        if (i+1)%m == 0:
            answer += (min(box)*m)
    
    return answer