HDU1424搬寝室

来源：互联网发布：大数据人工智能培训编辑：程序博客网时间：2024/05/22 02:01

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

状态Dp[i][j]为前i件物品选j对的最优解
当i=j*2时,只有一种选择即 Dp[i-2][j-1]+(w[i]-w[i-1])^2
当i>j*2时,Dp[i][j] = min(Dp[i-1][j],Dp[i-2][j-1]+(w[j]-w[j-1])^2)

#include<cstdio>#include<algorithm>using namespace std;#define min(x,y) (x)<(y)?(x):(y)int a[2005];int dp[2005][2005/2];void work(int n,int k){int i,j;dp[0][0] = 0;//当取0件的时候所耗费的精力为0for(i = 2; i <= n; i++){dp[i][0] = 0;for(j = 1; j <= i/2 ; j++)if(i -1 >= j*2)//当i-1>= j*2的时候dp[i][j] = min(dp[i-1][j],dp[i-2][j-1]+(a[i-1]-a[i])*(a[i-1]-a[i]));else//当i-1< j*2的时候dp[i][j] = dp[i-2][j-1] + (a[i-1]-a[i])*(a[i-1]-a[i]);}printf("%d\n",dp[n][k]);//输出最小的疲劳度。}int main(){int n,k,i;while(~scanf("%d%d",&n,&k)){for(i = 1; i <= n; i++)//输入数据，输入从1开始。scanf("%d",a+i);sort(a+1,a+n+1);//将元素从小到大进行排序。work(n,k);}return 0;}