//数直線上の点をK個の集合に分ける。Σmax(集合) - min(集合)を最小化 //・数直線上で各グループが連結になる場合だけ考えればよい。→そうじゃない場合は、より改善ができるため。 //・コストは、aMax - aMinから区間の連結部分K-1箇所の距離を引いたものになる。 //・連結部分は好きに選べるので、aを昇順ソートしたとすれば、aMax - aMin - 区間差上位K - 1個が答えになる。 #include #include #include #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 ans = a[n - 1] - a[0]; for (i = 0; i < k - 1; i++) ans -= b[i]; cout << ans << endl; return 0; }