hdoj 1421 搬寝室(DP)

题目大意：

在n件物品中选k对物品，每对物品的重量差的平方为疲劳度，求搬k对的最小总的疲劳度。

思路：

a[i]保存物品重量，需要先排序。因为任意一个数跟它相邻的数的的差最小。

b[i][j]保存前i件物品中取j对的最小疲劳度。

则 分两种情况

当 i-1>=j*2 时

b[i][j]=min{b[i-1][j] ,b[i-2][j-1]+(a[i]-a[i-1])^2} .

当 i-1<j*2 时

b[i][j]=b[i-2][j-1]+(a[i]-a[i-1])^2 .

 

#include <iostream>using namespace std;int a[2002],b[2002][1002];int get_min(int a,int b){if(a<b)return a;elsereturn b;}int cmp(const void *a,const void *b){return *(int *)a-*(int *)b;}int main(){ int n,k; int i,j,t1,t2; while(scanf("%d%d",&n,&k)!=EOF) {for(i=1;i<=n;i++)scanf("%d",&a[i]);qsort(a+1,n,sizeof(a[0]),cmp);memset(b,0,sizeof(b));b[2][1]=(a[2]-a[1])*(a[2]-a[1]);for(i=3;i<=n;i++){t1=i/2;t2=(a[i]-a[i-1])*(a[i]-a[i-1]);for(j=1;j<=t1 && j<=k;j++) if(i-1>=j*2)b[i][j]=get_min(b[i-1][j],b[i-2][j-1]+t2); elseb[i][j]=b[i-2][j-1]+t2; } printf("%d/n",b[n][k]);}return 0;}