[백준] 13164번 행복 유치원 (파이썬)

문제링크: https://www.acmicpc.net/problem/13164

문제해결

• 키 차이를 저장한 배열을 만들고
• 배열을 내림차순으로 정렬한 다음
• 큰 순서대로 k-1 개를 제외한 나머지의 합을 구하면 된다.

소스코드

import sys

input = sys.stdin.readline

n,k = map(int, input().split())

arr = list(map(int,input().split()))

if k == 1:
  print(max(arr) - min(arr))
  exit()

diff = []

for i in range(n-1):
  diff.append(arr[i+1]-arr[i])

diff.sort(reverse=True)

print(sum(diff[k-1:]))