链接:http://acm.hdu.edu.cn/showproblem.php?pid=1421
题目本就是中文的,题意在此不多说。
分析:这是一道典型的dp,每次的疲劳度等于左右手所提重物之差的平方。所以要尽量的使两手重物之差最小,所以首先想到应该先排序。dp[i][j]表示前i件物品中选择j对的疲劳度.这里的初始化要注意,有可能前i件里边一对都不选择,所以应该将dp[i][0]初始化为0.其他都初始化为最大值(大于2^15的平方即可,注意不要超了int 范围,我在这里栽了)。
状态转移方程:dp[i][j]=min{dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]),dp[i-1][j]} 是否取了第i件
代码:
#include<stdio.h>
#include<stdlib.h>
#define INF 1000000
int dp[2001][1010];//dp[i][j]表示前i件物品中选择j对物品的疲劳度
int mmin(int a,int b)
{
return a<b?a:b;
}
int comp(const void *a,const void *b)
{
return *(int *)a-*(int *)b;
}
int main()
{
int i,j,a[2001],n,k;
while(scanf("%d%d",&n,&k)!=EOF)
{
a[0]=0;
for(i=1;i<=n;i++)
scanf("%d",a+i);
qsort(a+1,n,sizeof(int),comp);
for(i=0;i<=n;i++)
for(j=1;j<=k;j++)
dp[i][j]=INF;
for(i=0;i<=n;i++)
dp[i][0]=0;//一对可以都不取
for(i=2;i<=n;i++)
for(j=1;j*2<=i;j++)
dp[i][j]=mmin(dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]),dp[i-1][j]);//是否取了第i件物品
printf("%d\n",dp[n][k]);
}
return 0;
}