时间限制: 1000 ms 空间限制: 262144 KB 具体限制
题目描述
在修建完新路后,小羊们总算可以安心入学了。今年是羊年,新入学的小羊特别多。老师们打算将N只小羊分成M个班级,每个班至少有1只羊。
如何分班成了老师们最头疼的事情,因为开学典礼上,村长就要看到小羊们列队的情况。每个班的小羊都排成一排,站在草场上。村长希望队列中羊的高度尽可能整齐,村长对队列的不整齐度有自己的要求。
例如队列中共有t只羊,高度依次为A1,A2……,At。那么不整齐度为:(|A1-A2|+|A2-A3|+……+|At-1-At|)2。即相邻两只羊高度差之和的平方。
而总体的不整齐度,就是各班不整齐度之和。
现在,请你帮助老师们设计一下,如何分班,如何列队,才能使M个班级的不整齐度之和最小。
输入
第一行两个整数N和M,分别表示共有N只小羊,要被分成M个班级。
第二行N个整数,表示每只小羊的高度Ai。
输出
输出最小的不整齐度之和,结果保证不会超过2^31-1。
样例输入
4 2
4 1 3 2
样例输出
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。
提示
分成两班,4和3一个班,1和2一个班,不管怎么排,两个班的不整齐度都是1,不整齐度之和为2。
来源/作者: 【宁波2015】
分析:这题80分的dp特别水:
设f[i][j] 为i只羊分了j个班的最小不整齐度。
那么f[i][j] = f[i - 1][k] + (a[i] - a[k + 1])^2
然后我们就可以拿到80分了,那怎么才能拿到100呢?斜率优化!
我们先把上面的式子co下来:f[i][j] = f[i - 1][k] + (a[i] - a[k + 1])^2
我们把后面那个式子用完全平方公式拆开:
f[i][j] = f[i - 1][k] + a[i]^2 + a[k + 1]^2 - 2 * a[i] * a[k + 1];
= (f[i - 1][k] + a[k + 1]^2 + a[i]^2) + a[i] * ((-2) * a[k + 1]) ;
是不是感觉很像一次函数?
设y=f[i][j],k=((-2) * a[k + 1]),x=a[i];
那么y=kx+b.
好了我们可以发现,x单调递增,k单调递减,那我们维护i个凸壳即可。