思路:用f[][]表示i件要搬j件的最小,主要在于要不要搬后一个
如果要,则getF(i-2,j-1)+a[i-1]
否则,getF(i-1,j)
公式 f[n][k] = min(f[n- 1][k], f[n - 2][k - 1] + a[n - 1])
getF用来处理一些边界情况,例如i必须比2*j大或等于,j不能=0
代码如下:
#include<stdio.h>
#include<algorithm>
using namespace std;
int f[2001][1001];//用于存i件的搬j的最小疲劳值
int a[2001];
int getF(int i, int j)
{
if(j * 2 > i) //要搬的数量超过物品总数时
return 1000000000;
if(j == 0) //一件物品都没搬时
return 0; //不消耗疲劳值
return f[i][j]; //正常情况
}
int main()
{
int n, k, i, j;
while(scanf("%d%d", &n, &k) != EOF)
{
for(i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
}
sort(a + 1, a + n + 1);
//a[i-1]就存了(a[i]-a[i-1])的平方
for(i = 1; i < n; ++i)
{
a[i] = a[i + 1] - a[i];
a[i] *= a[i];
}
for(j = 1; j <= k; ++j)
{
for(i = 2 * j; i <= n; ++i)
{
//要不要第i个
f[i][j] = min(getF(i - 1, j), getF(i - 2, j - 1) + a[i - 1]);
}
}
printf("%d\n", f[n][k]);
}
return 0;
}