結果

問題 No.615 集合に分けよう
コンテスト
ユーザー yomo3
提出日時 2020-03-23 00:58:42
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 346 bytes
コンパイル時間 294 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 19,988 KB
最終ジャッジ日時 2024-12-26 05:45:31
合計ジャッジ時間 3,726 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

from operator import itemgetter

N, K = map(int, input().split())
A = list(map(int, input().split()))

ans = 0
A.sort()
d = [A[i + 1] - A[i] for i in range(N - 1)]
d = list(zip(d, range(N - 1)))
d.sort(reverse=True, key=itemgetter(0))
s = 0
for i in range(K):
    e = -1 if i == K - 1 else d[i][1]
    ans += A[e] - A[s]
    s = e + 1
print(ans)
0