HDU 1421 搬寝室 二维DP

传送门：HDU 1421 搬寝室

分析：
先对n中物品的重量排序
令dp[i][j]表示前i个物品中选j对的最小疲劳度。
则dp[i][j]可能含有第i个物品(这种情况下,第i种物品一定是和第i-1个物品配对)，
则dp[i][j]=dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1])
dp[i][j]的j对也可能不含有第i个物品，此时有
dp[i][j]=dp[i-1][j]
状态转移方程
dp[i][j]=min{dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1]),dp[i-1][j]

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define ms(x, y) memset(x, y, sizeof(x))
const double PI = acos(-1.0), eps = 1e-8;
#define size 2005
#define INIT 2147483646
int dp[size][1005];
int tired[2002];
int main() {
    int n,k,i,j;
    int val[size] = {0};
    dp[0][0] = 0;
    while (scanf("%d%d",&n,&k) != EOF) {
        val[0] = 0;
        for(i=1;i<=n;i++)
            scanf("%d",val+i);
        // 预处理：按疲劳值 从小到大进行排序
        sort(val+1, val+n+1);
        // 计算排序后相邻两个的疲劳值
        for(i=2; i<=n; i++)
            tired[i] = (val[i]-val[i-1])*(val[i]-val[i-1]);
        for (i=0;i<=n;i++)
            for(j=1;j<=k;j++)
                dp[i][j] = INIT;

        for (i=2; i<=n; i++)
            for (j=1; j*2<=i; j++) // j代表搬动的次数 1次班搬动两个物品
                dp[i][j] = min(dp[i-2][j-1]+tired[i], dp[i-1][j]); // dp[i-1][j] 不能单独搬动一个物品

        printf("%d\n",dp[n][k]); // 前n件商品中 , k次搬动(k对物品)的最小疲劳值
    }
    return 0;
}