HDU 1421 搬寝室

来源:互联网 发布:怎么弄大批量淘宝号 编辑:程序博客网 时间:2024/04/27 21:45

最近学习动态规划,总是找不出转移方程。。。

  这里的是:    if(i==2*j) dp[i][j]=dp[i-2][j-1]+pow2(a[i]-a[i-1]);
                          else if(i>2*j)  dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+pow2(a[i]-a[i-1]));  

#include<stdio.h>
#include<math.h>
#include<memory.h>
int a[2100];
int dp[2100][1100];
int  min(int a,int b)
{
 return a < b ? a : b;
}

int  pow2(int a)
{
 return   a*a;
}

int cmp(const void *a,const void *b)
{
    return  *(int *)a - *(int *)b;
}
int main()
{
 int i,j,k,n,l;
    while(scanf("%d %d",&n,&k)!=EOF)
 {
  for(i=1;i<=n;i++)  scanf("%d",&a[i]);
  qsort(a+1,n,sizeof(a[1]),cmp);
     for(i=0;i<=n;i++)
         for(j=0;j<=k;j++)
             dp[i][j]=0;
     for(i=2;i<=n;i++){
   l=i/2;
         for(j=1;j<=l;j++){
          if(i==2*j) dp[i][j]=dp[i-2][j-1]+pow2(a[i]-a[i-1]);
    else if(i>2*j)  dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+pow2(a[i]-a[i-1]));  
   }
  }
  printf("%d\n",dp[n][k]);
 }
 system("pause");
 return 0; 
}