이코테 | 그리디 - 큰 수의 법칙 [Python]

💡 Python 3

🔗 문제

큰 수의 법칙

💻 코드

# N, M, K = 5, 8, 3
# num_list = [2, 4, 5, 4, 6]
N, M, K = map(int, input().split())
num_list = map(int, input().split())

def solution(N, M, K, num_list):
    result = 0
    num_list.sort()
    for i in range(1, M+1):
        if i % K != 0:
            result += num_list[-1]
        else:
            result += num_list[-2]
    return result

print(solution(N, M, K, num_list))

🧩 깨달음

파이썬 스터디에서 같이 푼 문제인데 .. 민소님의 코드를 보고 완전 감탄했다!!
민소님은 수학적으로 접근해서 몫, 나머지, sum()를 활용해서 for문 없이 문제를 푸셨다.
참고해야지!!