有 n 名学生参加军训,军训的一大重要内容就是走队列,而一个队列的不整齐程度是该队中最高的学生的身高与最矮的学生的身高差值的平方。
现在要将 n 名参加军训的学生分成 k 个队列,每个队列的人数可以是任意非负整数。
在安排队列时希望所有队列的不整齐度之和尽量小,请问不整齐度之和最小可以是多少?
输入格式
第一行两个整数 n,k,表示学生人数和队列数。
第二行 n 个整数,依次表示每名学生的身高。
输出格式
一个整数表示答案。
数据范围
对于10%的数据,k=1;
对于另外10%的数据,k=2;
对于另外10%的数据,k=3;
对于另外10%的数据,k=4;
对于另外10%的数据,1≤n,k≤5;
对于另外10%的数据,1≤n,k≤10;
对于另外20%的数据,1≤n,k≤100;
对于另外5%的数据,n=k=500
对于所有的数据,1≤n,k≤500,0≤学生身高 ≤200
输入样例:
3 2
170 180 168
输出样例:
4
这道题类似于 矩阵连乘,但是又不同于矩阵连乘的分组,因为这里限制了分组的数量,而矩阵连乘没有限制分组的数量
可以证明,将分组用完一定不会使得平方和变大,可能变小也可能不变,因为如果有空的分组的话,挪一个其他分组的过去,只有一个元素的分组的平方和是0,并不会增大结果
f[i][j] 代表的是,前i个元素分成j组的最小结果(平方和)
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int h[510];
int f[510][510];
int main(){
int n,m;
cin>>n>>m;
if (n <= m){
cout<<0;
return 0;
}
for (int i = 1; i <= n; ++i)
cin>>h[i];
sort(h + 1,h + n + 1);
memset(f,0x3f,sizeof f);
f[0][0] = 0;
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j){
if(j >= i){
f[i][j] = 0;
continue;
}
for (int k = i; k >= j; --k){ // f[i][j]将i个人分成j份,每次将f[i][j] 分成 f[k - 1][j - 1] 和 [k,...,i] 两部分
f[i][j] = min(f[i][j],f[k - 1][j - 1] + (h[i] - h[k]) * (h[i] - h[k]));
}
}
cout<<f[n][m];
return 0;
}