• 출처 : https://www.acmicpc.net/problem/2212

• 설명 : 모든 센서를 k개의 구간으로 나누는 것과 동일

• 풀이

  1. 센서 리스트 정렬
  2. 모든 센서들 간의 거리 set
  3. 모든 센서들 간의 거리 정렬
  4. k-1번 pop

• 참고 : https://journeytosth.tistory.com/16

코드

def init():
    n = int(input(''))
    k = int(input(''))
    sensor_list = list(map(int, input('').split(' ')))
    return n, k, sensor_list

n, k, sensor_list = init()

sensor_list.sort()

# 모든 센서들 간의 거리 저장
dist_list = []
for i in range(n-1):
    dist_list.append(sensor_list[i+1] - sensor_list[i])

dist_list.sort()

# 큰 순서로 k-1개 삭제
for i in range(k-1):
    if len(dist_list) > 0:
        dist_list.pop()

print(sum(dist_list))