백준 13164번(Python) : 행복 유치원

www.acmicpc.net/problem/13164

13164번: 행복 유치원

---

[풀이]

 

 리스트에 인접한 두 사람의 키의 차이를 리스트에 저장을 한 후, 정렬을 하였습니다. 이후, N명의 사람을 K조로 나누었으므로 (N-K)까지의 키 차이를 더하면 값을 얻을 수 있다.

 

---

[코드]

 

import sys
input = sys.stdin.readline

n,k = map(int,input().split())
arr = list(map(int,input().split()))
result = 0
temp = []
for i in range(1,n):
    temp.append(arr[i] - arr[i-1])
temp.sort()

if k == n:
    result = 0
    print(result)
else:
    for i in range(n-k):
        result += temp[i]
    print(result)