[백준] 20186번 수 고르기

💡문제접근

• 오름차순 정렬을 수행한 후 뒤에서부터 K개의 수를 골라서 조건대로 점수를 계산해주면 된다.

💡코드(메모리 : 31256KB, 시간 : 56ms)

import sys
input = sys.stdin.readline

N, K = map(int, input().strip().split())
arr = list(map(int, input().strip().split()))
arr.sort()

Sum = 0
t = 0
for i in range(N-K, N):
    Sum += (arr[i] - t)
    t += 1
print(Sum)

💡소요시간 : 8m