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;
}