HDU 1421 搬寝室

【题目链接】
http://acm.hdu.edu.cn/showproblem.php?pid=1421

题目意思

xhd要搬宿舍，有n件物品，只搬其中2*k件，每次搬运时疲劳值值为左右手的物品重量差的平方。问最少用多疲劳值搬完。

解题思路

不难想到相邻的物品一起搬最省力，所以将n件物品排序，从第二件物品开始没次加一件物品，如果选择搬运就和前一个物品一起搬，或者选择不搬。所以状态转移方程为：dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));///i表示第几个物品，j表示搬了多少趟（k）。值的注意的初始化不够大会wa

代码部分

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn=1e4+5;
int dp[2005][2005];
int a[2005];
int main()
{
    int n,k;
    while(cin>>n>>k)
    {
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                dp[i][j]=2147483346;  ///注意初始要大
        dp[0][0]=0;
        for (int i=1;i<=n;i++)
            cin>>a[i];
        sort(a+1,a+n+1);
        for (int i=2;i<=n;i++)  
            for (int j=1;j<=k&&j*2<=i;j++)
                dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
        cout<<dp[n][k]<<endl;
    }
    return 0;
}