📚 13164 - 행복 유치원

행복 유치원

이해

문제에 대한 로직이 떠오르지 않아 구글링 했다.

참고 블로그

1 3 5 6 10 일때

2 2 1 4 만큼 거리가 차이난다.

여기서 내림차순 정렬 후, k-1 인덱스에서 마지막 인덱스까지 더해주면 된다. (티셔츠 비용)

소스

import sys  
  
read = sys.stdin.readline  
  
n, k = map(int, read().split())  
  
arr = list(map(int, read().split()))  
  
s = [0] * (n + 1)  
  
for i in range(n - 1):  
    s[i] = arr[i + 1] - arr[i]  
  
s.sort(reverse=True)  
  
answer = 0  
  
for i in range(k - 1, n):  
    answer += s[i]  
  
print(answer)