Hdu 1421 搬寝室

Hdu 1421 搬寝室(dp)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1421

解题思路:
对于每一位的物品,都有取与不取两种做法，如果取了，那就要加上疲劳值，不取，那就等于前面算好的结果，(初始化要完整)，用二维dp来处理，i表示物品数，j表示物品对数，状态转移方程:dp[i]j=min(dp[i-1]j,dp[i-2][j-1]+(num[i] - num[i - 1]) * (num[i] - num[i - 1])(取的话就相当于物品数减少2的情况下再加上取得这两个物品产生的疲劳值))。
代码如下:

#include <bits/stdc++.h>

#define inf 0x3f3f3f
#define MAXN 2005
int dp[MAXN][MAXN], num[MAXN];

inline int min(int a, int b) {//取小
    return a > b ? b : a;
}

int main() {
    int n, k;
    while (scanf("%d %d", &n, &k) != EOF) {
        memset(dp, inf, sizeof(dp));
        dp[0][0] = 0;//i为0是第一次dp跑的位置，不能忘记清0
        for (int i = 1; i <= n; i++) {
            scanf("%d", &num[i]);
            dp[i][0] = 0;
        }
        std::sort(num + 1, num + n + 1);//排序保证相邻最小
        for (int i = 2; i <= n; i++) {
            for (int j = 1; j <= i / 2; j++) {
                dp[i][j] = min(dp[i - 1][j], dp[i - 2][j - 1] + (num[i] - num[i - 1]) * (num[i] - num[i - 1]));//取两种方案的最小值
            }
        }
        printf("%d\n", dp[n][k]);
    }
    return 0;
}