입출력 예시
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 |
나의 코드
def solution(k, m, score):
highest_batch = sorted(score, reverse=True)
answer = 0
for i in range(len(score) // m):
answer += highest_batch[m*(i+1)-1] * m
return answer
Python
복사
다른 풀이
def solution(k, m, score):
return sum(sorted(score)[len(score)%m::m])*m
Python
복사
개선점 분석
•
“다른 풀이” 처럼 좀 더 짧고 간결하게 만들어보자.
•
reverse sort 하지 않고 하는게 더 깔끔.