NOIP宁波C组---羊羊队列

羊羊队列

题目描述

羊羊列队(queue.pas/c/cpp)

【问题描述】

在修建完新路后，小羊们总算可以安心入学了。今年是羊年，新入学的小羊特别多。老师们打算将N只小羊分成M个班级，每个班至少有1只羊。

如何分班成了老师们最头疼的事情，因为开学典礼上，村长就要看到小羊们列队的情况。每个班的小羊都排成一排，站在草场上。村长希望队列中羊的高度尽可能整齐，村长对队列的不整齐度有自己的要求。

例如队列中共有t只羊，高度依次为A1，A2……，At。那么不整齐度为：(|A1-A2|+|A2-A3|+……+|At-1-At|)。即相邻两只羊高度差之和。 而总体的不整齐度，就是各班不整齐度之和。

现在，请你帮助老师们设计一下，如何分班，如何列队，才能使M个班级的不整齐度之和最小。

【输入】

第一行两个整数N和M，分别表示共有N只小羊，要被分成M个班级。

第二行N个整数，表示每只小羊的高度Ai。

【输出】

输出最小的不整齐度之和，结果保证不会超过231-1。

【输入样例】

4 2 4 1 3 2

【输出样例】

2 【样例解释】

分成两班，4和3一个班，1和2一个班，不管怎么排，两个班的不整齐度都是1，不整齐度之和为2。

【数据范围】

30%的数据，1<=N<=10；1<=M<=5；

80%的数据，1<=N<=300；1<=Ai<=1000；

100%的数据，1<=N<=10000,1<=M<=1000，1<=Ai<=1000000，保证M<=N。

---

#include <bits/stdc++.h>
using namespace std;
int n,m,a[10010],b[10100],ans;
bool cmp (int x,int y) {
	return x<y;
}
int main() {
	cin>>n>>m;
	for (int i=1;i<=n;i++) 
	  scanf ("%d",&a[i]);
	sort (a+1,a+n+1,cmp);
	for (int i=2;i<=n;i++) 
	  b[i-1]=abs (a[i]-a[i-1]);
	sort (b+1,b+n,cmp);
	for (int i=1;i<=n-m;i++) ans+=b[i];
	cout<<ans;
	return 0;
}