JZOJ 初中 2184：羊羊列队

时间限制: 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个凸壳即可。