hdu 1421 搬寝室

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1421

这是道很久以前的遗留问题了，当时想的那种方法没问题就是老wa，就放下了。 昨晚就又用那种方法了，结果wa，队友指出了这种方法的错误之处，现在想想确实是不全面的。

这是到dp题，dp方程是用dp[i][j]表示从前i个人中选出来j对时的最小值，dp[i][j]=min{dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1])} 前面一个表示不选第i个数，后面表示选第i个数。

View Code

#include <iostream>
#include <algorithm>
using namespace std;
int main( )
{
int n,k,i,a[2002],sum[2002],b[2002],l,m,max,max1;
while (cin>>n>>k)
{
for (i=0;i<n;i++)
{
cin>>a[i];
}
sort(a,a+n);
m=max=max1=0;
l=0;
for (i=1;i<n;i+=2)
{
sum[m++]=(a[i]-a[i-1])*(a[i]-a[i-1]);
}
for (i=2;i<n;i+=2)
{
b[l++]=(a[i]-a[i-1])*(a[i]-a[i-1]);
}
sort(sum,sum+m);
sort(b,b+l);

for (i=0;i<k;i++)
{
max1+=b[i];
}
for (i=0;i<k;i++)
{
max+=sum[i];
}
if(n!=k*2){max=(max>max1)?max1:max;
cout<<max<<endl;}
else cout<<max<<endl;
}
return 0;
}