题意非常简单,一道中文题。
首先将所有重量进行排序,我们容易证明,两个相邻的数的差平方一定比其他位置的差平方要小。
那么dp方程就好写了,设dp[i][j]表示从i个物品里选出j对最小的疲劳度。则dp方程如下:
dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(weigh[i]-weigh[i-1])^2)
只考虑最后一个物品,如果他不能被选上,就相当于从前i-1个物品里面继续选j对,
否则如果这个物品被选上了,根据前面的证明,只有可能和他配对的是第i-1个,于是就相当于从前i-2个里面选j-1对,再加上新的那一对疲劳度。
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;
#define My_max 2002
#define INF 1073741824
int n,k;//从n个物品中选出k对使满足题意
int weigh[My_max];
int dp[My_max][My_max/2];
void DP()
{
for(int i=0;i<=n;i++)
for(int j=1;j<=k;j++)
dp[i][j]=INF;
dp[0][0]=0;
for(int i=2;i<=n;i++)
for(int j=1;j<=i/2;j++)
dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(weigh[i]-weigh[i-1])*(weigh[i]-weigh[i-1]));
}
int main()
{
while(scanf("%d %d",&n,&k)!=EOF)
{
memset(weigh,0,sizeof(weigh));
for(int i=1;i<=n;i++)
scanf("%d",&weigh[i]);
sort(weigh+1,weigh+n+1);
DP();
printf("%d\n",dp[n][k]);
}
return 0;
}