HDU1421 搬寝室 动态规划

动态规划题

题意是输入两个整数n和k，表示从n件物品中搬走k对（每次左右手各搬一件），已知每件物品的疲劳值，求搬走k对后的最小疲劳度。

每个物品的疲劳度用ans存起来，很明显，把ans按升序排序后每相邻两个物品的疲劳度都是最小的；

然后定义数组dp【i】【j】表示从前i件物品中搬走j对的总疲劳度，则对于第i件物品：

（1）如果搬走第i件，那么一定也要搬走第i-1件物品，不然第i件物品和谁配对呢，同时从前i-2件物品中搬走j-1对，

          此时dp【i】【j】=dp【i-2】【j-1】+（ans【i】-ans【i-1】）^2;

（2）如果不搬走第i件物品，那么就等于从前i-1件物品里面搬走j对物品了，即dp【i】【j】=dp【i-1】【j】;

显然状态转移方程：dp【i】【j】=min（dp【i-2】【j-1】+（ans【i】-ans【i-1】）^2，dp【i-1】【j】）；

#include <cstdio>

#include <algorithm>
#include <iostream>
#define MAX 2100
#define INF 999999999
using namespace std;
int ans[MAX],dp[MAX][MAX/2];
int main()
{
    int n,k;
    while(cin>>n>>k)
    {
        for(int i=1;i<=n;i++)
          scanf("%d",&ans[i]);
        sort(ans+1,ans+1+n);
        for(int i=1;i<=n;i++)
          for(int j=0;j<=k;j++)
            dp[i][j]=INF;
        for(int i=0;i<=n;i++)
          dp[i][0]=0;
        for(int i=2;i<=n;i++)
          for(int j=1;j<=k&&2*j<=i;j++)
            dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(ans[i]-ans[i-1])*(ans[i]-ans[i-1]));
        cout<<dp[n][k]<<endl;
    }
    return 0;
}