結果

問題 No.615 集合に分けよう
ユーザー startcpp
提出日時 2017-12-29 20:25:32
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 931 bytes
コンパイル時間 616 ms
コンパイル使用メモリ 64,756 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-21 10:35:10
合計ジャッジ時間 2,343 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

//数直線上の点をK個の集合に分ける。Σmax(集合) - min(集合)を最小化
//・数直線上で各グループが連結になる場合だけ考えればよい。→そうじゃない場合は、より改善ができるため。
//・コストは、aMax - aMinから区間の連結部分K-1箇所の距離を引いたものになる。
//・連結部分は好きに選べるので、aを昇順ソートしたとすれば、aMax - aMin - 区間差上位K - 1個が答えになる。
#include <iostream>
#include <algorithm>
#include <functional>
#define int long long
using namespace std;

int n, k;
int a[50000];
int b[49999];

signed main() {
	int i;
	
	cin >> n >> k;
	for (i = 0; i < n; i++) cin >> a[i];
	sort(a, a + n);
	for (i = 0; i < n - 1; i++) b[i] = a[i + 1] - a[i];
	sort(b, b + n, greater<int>());

	int ans = a[n - 1] - a[0];
	for (i = 0; i < k - 1; i++) ans -= b[i];
	cout << ans << endl;
	return 0;
}
0