题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1421
思路:
先排序,每次依次选取相邻的两个,可保证消耗的体力值最小。
dp[i][j]表示在前i个物品中选取j对物品所消耗的体力值。
由于每次取一对,即2件物品,前i件物品可能会出现:
1)i = j * 2,此时取的物品正好是i - 1和i,有dp[i][j] = dp[i - 2][j - 1] + ( v[i] - v[i - 1] )^2,v[]表示物品重量。
2)i > j * 2,此时表示奇数个物品,最后一个物品i可以选择也可以不选择,有dp[i][j] = dp[i - 1][j],(j对物品所以是j)
所以状态转移方程:
dp[i][j] = min {dp[i - 2][j - 1] + ( v[i] - v[i - 1] )^2,dp[i][j] = dp[i - 1][j]}
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define SIZE 2005
using namespace std;
typedef __int64 Int;
int n,k;
Int v[SIZE];
Int dp[SIZE][SIZE];
int main()
{
while(~scanf("%d%d",&n,&k))
{
for(int i=1; i<=n; i++)
scanf("%I64d",&v[i]);
sort(v+1,v+1+n);
for(int i=0; i<=n; i++)
for(int j=0; j<=k; j++)
dp[i][j] = 0;
for(int i=2; i<=n; i++)
{
for(int j=1; j*2<=i; j++)
{
if(i == 2*j)
dp[i][j] = dp[i-2][j-1]+(v[i]-v[i-1])*(v[i]-v[i-1]);
else
dp[i][j] = min(dp[i-1][j],dp[i-2][j-1]+(v[i]-v[i-1])*(v[i]-v[i-1]));
}
}
printf("%I64d\n",dp[n][k]);
}
return 0;
}
code reference: http://www.cnblogs.com/gentleh/archive/2013/04/05/3000746.html