HDU 1421 搬寝室 DP

http://acm.hdu.edu.cn/showproblem.php?pid=1421

题目：

搬寝室是很累的(T T ,这学期才搬寝室），xhd要从n件物品中搬走2*k件，（左右手各拿一件），他的每次疲劳度和左右手的物品的重量差的平方成正比，求搬完后最小的疲劳度。

设dp[i][j]为i件物品中选择j对的最小疲劳度。则若包含第i件物品则dp[i][j]=dp[i-2][j-1] * (weight[i]-weight[i-1])*(weight[i]-weight[i-1])

若不包含第i件则dp[i][j]=dp[i-1][j]。

则 dp[i][j]= min (dp[i-1][j], dp[i-2][j-1] * (weight[i]-weight[i-1])*(weight[i]-weight[i-1]) );

还有初始化的问题。。。

j=0的时候应该是0啊啊啊啊啊。。。WA了一次

#include<cstdio>
#include<algorithm>
using namespace std;
const int INF=INT_MAX;
const int MAXN=2000+10;
int weight[MAXN];
int dp[MAXN][MAXN>>1]={0};
inline int getweight(int i)
{
	return (weight[i]-weight[i-1])*(weight[i]-weight[i-1]);
}
int main()
{
	int n,k;
	while(~scanf("%d%d",&n,&k))
	{
		for(int i=1;i<=n;i++)
			scanf("%d",&weight[i]);

		sort(weight+1,weight+n+1);

		for(int i=0;i<=n;i++)
			for(int j=1;j<=k;j++)
				dp[i][j]=INF;

		for(int i=2;i<=n;i++)
		{
			//dp[i][j] 表示前i件物品取 j 对
			for(int j=1;j+j<=i;j++)
				dp[i][j]= min (dp[i-1][j], dp[i-2][j-1] + getweight(i));
		}

		printf("%d\n",dp[n][k]);
	}

	return 0;	
}