首页 > 代码库 > hdu 1421 搬寝室

hdu 1421 搬寝室

http://acm.hdu.edu.cn/showproblem.php?pid=1421

 1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 using namespace std; 5  6 int dp[2011][2001]; 7 int a[2011]; 8 int n,k; 9 10 int sqr(int x)11 {12     return x*x;13 }14 15 bool cmp(const int a,const int b)16 {17     return a>b;18 }19 20 int main()21 {22     while(scanf("%d%d",&n,&k)!=EOF)23     {24         memset(dp,0,sizeof(dp));25         memset(a,0,sizeof(a));26         for(int i=1; i<=n; i++)27         {28             scanf("%d",&a[i]);29         }30         sort(a+1,a+n+1,cmp);31         for(int i=1; i<=k; i++)32         {33             dp[i][i*2]=dp[i-1][i*2-2]+sqr(a[i*2-1]-a[i*2]);34             for(int j=i*2+1; j<=n; j++)35             {36                 dp[i][j]=min(dp[i][j-1],dp[i-1][j-2]+sqr(a[j-1]-a[j]));37             }38         }39         printf("%d\n",dp[k][n]);40     }41     return 0;42 }
View Code