4111. 무선 단속 카메라 D4 파이썬

문제링크 : https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWJHjcFqdyoDFAUH&categoryId=AWJHjcFqdyoDFAUH&categoryType=CODE

 

SW Expert Academy

 

풀이 : 경로가 긴 것부터 제거해준다. 제거할 수 있는 개수는 카메라의 개수 -1

T = int(input())
# 여러개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
for test_case in range(1, T + 1):
    N= int(input())
    K= int(input())
    cameralist = sorted(list(set(map(int,input().split()))))
    if K >= len(cameralist):
        print('#{} {}'.format(test_case, 0))
        continue
    dist = list()
    for i in range(len(cameralist)-1):
        dist.append(cameralist[i+1]-cameralist[i])
    dist.sort()
    while K!=1:
        dist.pop()
        K-=1
    print('#{} {}'.format(test_case, sum(dist)))