对于DP题目, 一直没能入门, 每次都不能找到正确的状态转移方程。
借鉴别人代码
dp[i][j] i 代码i件物品, j 代表 组成 j 对时的最小疲劳值。
dp[i][j] = min(dp[i-1][j] , dp[i-2][j-1] + (value[i-1] - value[i])*(value[i-1] - value[i]));
解释: 如果当前 第 i 件物品不选, 则dp[i][j] 由 dp[i-1][j] 得到, 若选第i 件物品, 则 必选 i - 1 件物品,因为value[i]是排序后的序列, 只有 value[i-1] 和value[i]的差的平方和才满足最小条件,故得到状态转移方程式;
借鉴别人代码
dp[i][j] i 代码i件物品, j 代表 组成 j 对时的最小疲劳值。
dp[i][j] = min(dp[i-1][j] , dp[i-2][j-1] + (value[i-1] - value[i])*(value[i-1] - value[i]));
解释: 如果当前 第 i 件物品不选, 则dp[i][j] 由 dp[i-1][j] 得到, 若选第i 件物品, 则 必选 i - 1 件物品,因为value[i]是排序后的序列, 只有 value[i-1] 和value[i]的差的平方和才满足最小条件,故得到状态转移方程式;
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int M = 2105;
const int inf = 99999999;
int dp[M][M];
int value[M];
int n, m;
int main()
{
while(scanf("%d%d", &n, &m) != EOF) {
value[0] = 0;
for(int i = 1; i <= n; i++)
scanf("%d", &value[i]);
sort(value +1, value + n + 1);
dp[0][0] = 0;
for(int i = 0; i <= n; i++)
for(int j = 1; j <= m; j++)
dp[i][j] = inf;
for(int i = 2; i <= n; i++){
for(int j = 1; j*2 <= i; j++)
dp[i][j] = min(dp[i-1][j], dp[i-2][j-1] + (value[i]-value[i-1])*(value[i]-value[i-1]));
}
printf("%d\n", dp[n][m]);
}
return 0;
}