hdu 1421 搬寝室

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
# define MAX 2005
using namespace std;

int n,k;
int a[MAX];
int dp[MAX][MAX];   //前i个物品中选择j对产生的最小疲劳度

int value(int i,int j)
{
return (i-j) * (i-j);
}

void solve()
{
int i,j;
for(i=2; i<=n; i++)
{
for(j=1; j<=k; j++)
{
if(i == 2 * j)
dp[i][j] = dp[i-2][j-1] + value(a[i],a[i-1]);
else
dp[i][j] = min(dp[i-2][j-1] + value(a[i],a[i-1]), dp[i-1][j]);
}
}
}
int main()
{
int i;
while(cin >> n >> k)
{
for(i=1; i<=n; i++)
scanf("%d",&a[i]);
sort(a+1,a+n+1);
memset(dp,0,sizeof(dp));
solve();
printf("%d\n",dp
[k]);
}
return 0;
}