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个数。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int INF=123456178;
int dp[2002][1005];
int a[2005];
int min(int x,int y)
{
    if(x>y) return y;
    else return x;
}
int main()
{
    int i,j,k,n;
    while (scanf("%d%d",&n,&k)!=EOF)
    {
        for (i=1;i<=n;i++)
            scanf("%d",&a[i]);
        
        for (i=0;i<=n;i++)   //初始化
        {
            for(j=1;j<=k;j++)
                dp[i][j]=INF;
            dp[i][0]=0;
        }
        sort(a+1,a+1+n);    //排序
        for (i=2;i<=n;i++) 
            for (j=1;j<=k;j++)
            {
                dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
            }
            if(n==1)printf("0\n");
            else 
            printf("%d\n",dp[n][k]);
    }
    return 0;
}

下面的wa

#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;
}