hdu 1421 搬寝室

/*排序后最佳的选择一定是相邻的*/
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int dp[2005][2005];
int a[2005];
int main()
{
    int n,k;
    while(scanf("%d %d",&n,&k)==2)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        sort(a+1,a+1+n);

        for(int i=1;i<=n;i++)  /*我不会告诉你这样初始化能快400ms*/
        memset(dp[i],0x3f,sizeof(int)*(n+1));

   //   memset(dp,0x3f,sizeof(dp));

        for(int i=0;i<=n;i++) dp[i][0]=0;
        for(int i=2;i<=n;i++)
        {
            for(int j=1;2*j<=i&&j<=k;j++)
            {
                dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i-1]-a[i])*(a[i-1]-a[i]));
            }
        }
        printf("%d\n",dp[n][k]);
    }
    return 0;
}