hdu1421_搬寝室

题目：搬寝室

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

 #include<stdio.h>
 #include<algorithm>
 using namespace std;
 #define N 2100
 typedef long long LL;
 LL dp[N][N/];        //dp[i][j]: 前 i 件中拿走j对最少疲劳度
 LL a[N];              //记录重量
 LL fun(LL i,LL j)     //一次性搬运第 i 件和第 j 件消耗的疲劳度
 {
   return (a[j]-a[i])*(a[j]-a[i]);
 }
 LL min(LL a,LL b)
 {
   return a>b?b:a;
 }
 int main()
 {
   LL n,k;
   while(scanf("%I64d%I64d",&n,&k)!=EOF)
   {
     for(LL i=;i<=n;i++)
     {
       scanf("%I64d",&a[i]);
     }
     sort(a,a+n+);  //关键的一步，排序，排完序以后每次拿走的某对就肯定是相邻的
     for(LL i=;i<=n;i++)
       for(LL j=;j<=n/;j++)
         dp[i][j]=;
     for(LL i=;i<=n;i++)  //不管前几件，拿走0对疲劳度肯定为0
       dp[i][]=;
     dp[][]=fun(,);    //前两件取一对疲劳度固定
     for(LL i=;i<=n;i++)
     {
       for(LL j=;j<=i/;j++)
       {
         dp[i][j]=min(dp[i-][j],dp[i-][j-]+fun(i-,i));
         //前i件取j件分两种情况：
         //1.第i件不取：那么最佳情况为dp[i-1][j];
         //2.第i件取走：那么第j件必然和第j-1件一起拿，那么最佳为dp[i-2][j-1]+fun(i-1,i)
       }
     }
     printf("%I64d\n",dp[n][k]);
   }
   return ;
 }