AizuOJ 2104 Country Road（排序）

Country Road
Time Limit : 5 sec, Memory Limit : 131072 KB
Japanese
Problem E: カントリーロード
ある過疎地域での話である． この地域ではカントリーロードと呼ばれるまっすぐな道に沿って， 家がまばらに建っている． 今までこの地域には電気が通っていなかったのだが， 今回政府からいくつかの発電機が与えられることになった． 発電機は好きなところに設置できるが， 家に電気が供給されるにはどれかの発電機に電線を介してつながっていなければならず， 電線には長さに比例するコストが発生する． 地域で唯一の技術系公務員であるあなたの仕事は， すべての家に電気が供給されるという条件の下で， できるだけ電線の長さの総計が短くなるような発電機 および電線の配置を求めることである． なお，発電機の容量は十分に大きいので， いくらでも多くの家に電気を供給することができるものとする．

サンプル入力の1番目のデータセットを図2に示す． この問題に対する最適な配置を与えるには， 図のように x = 20 と x = 80 の位置に発電機を配置し， それぞれ図中のグレーで示した位置に電線を引けばよい．

発電機と電線の配置の例（入力例の最初のデータセット）．
図2: 発電機と電線の配置の例（入力例の最初のデータセット）．
Input
入力の1行目にはデータセットの個数 t (0 < t ≤ 50) が与えられる．

引き続き t 個のデータセットが与えられる． データセットはそれぞれ次のような形式で2行で与えられる．

n k
x1 x2 ... xn

n は家の戸数，k は発電機の個数である． x1, x2, …, xn はそれぞれ家の位置を表す一次元座標である． これらの値はすべて整数であり， 0 < n ≤ 100000, 0 < k ≤ 100000, 0 ≤ x1 < x2 < … < xn ≤ 1000000 を満たす．

さらに，データセットのうち 90% は， 0 < n ≤ 100, 0 < k ≤ 100 を満たしている．

Output
各データセットに対し， 必要な電線の長さの総計の最小値を1行に出力せよ．

Sample Input
6
5 2
10 30 40 70 100
7 3
3 6 10 17 21 26 28
1 1
100
2 1
0 1000000
3 5
30 70 150
6 4
0 10 20 30 40 50
Output for the Sample Input
60
13
0
1000000
0
20
题目链接：https://onlinejudge.u-aizu.ac.jp/problems/2104
题目大意：在一个偏远地方、有n栋房子和k个发电机、问如何装发电机使得所需要的电线长度最小
solution：保存每个房屋之间的距离、排一个序、取前n-k小的区间之和就是答案

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int t, n, k, pre, x;
	scanf("%d", &t);
	while (t--){
		scanf("%d%d", &n, &k);
		vector<int> dis;
		for (int i = 0; i < n; ++i){
			if (!i)scanf("%d", &pre);
			else {
				scanf("%d", &x);
				dis.push_back(x - pre);
				pre = x;
			}
		}
		sort(dis.begin(), dis.end());
		if (k < n)printf("%d\n", accumulate(dis.begin(), dis.begin() + n - k, 0));
		else printf("0\n");
	}
	return 0;
}