結果
問題 |
No.615 集合に分けよう
|
ユーザー |
![]() |
提出日時 | 2018-03-31 00:24:11 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 164 ms / 2,000 ms |
コード長 | 695 bytes |
コンパイル時間 | 113 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 19,948 KB |
最終ジャッジ日時 | 2024-06-26 01:03:10 |
合計ジャッジ時間 | 3,647 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 26 |
ソースコード
# -*- coding: utf-8 -*- """ No.615 集合に分けよう https://yukicoder.me/problems/no/615 """ import sys from sys import stdin input = stdin.readline def solve(N, K, numbers): numbers.sort() gaps = [] for i in range(1, N): gaps.append((-(numbers[i] - numbers[i-1]), i)) gaps.sort() gaps = gaps[:K-1] ans = 0 b = 0 for gap, pos in gaps: ans += numbers[pos-1] - numbers[b] b = pos ans += numbers[-1] - numbers[b] return ans def main(args): N, K = map(int, input().split()) numbers = [int(x) for x in input().split()] ans = solve(N, K, numbers) print(ans) if __name__ == '__main__': main(sys.argv[1:])