HDU 1421 搬寝室 （dp）

#include<cstring>
#include<iostream>
#include<algorithm>
#define P(x,y) ((x-y)*(x-y))
using namespace std;
int a[2010],d[2010][2010];
int main()
{
int i,j,n,k;
while(cin>>n>>k){
for(i=0;i<n;++i)
cin>>a[i];
sort(a,a+n);
memset(d,0,sizeof(d));
for(i=1;i<=n;++i)
for(j=1;j<=k;++j)
if(i>2*j){
d[i][j]=min(d[i-1][j],d[i-2][j-1]+P(a[i-1],a[i-2]));
}else{
d[i][j]=d[i-2][j-1]+P(a[i-1],a[i-2]);
}
cout<<d
[k]<<endl;
}
system("pause");
return 0;
}