HDU 1421 搬寝室【DP】

题目链接

题目意思

现在要搬寝室，总共有n件物品，现在要任意搬2*k件物品过去。每次搬东西都有一个疲劳度。这个疲劳度就是右手重量减去左手重量的平方。例如左手搬重量为3的物品，右手搬重量为6的物品，那么疲劳度就是（6-3）^2=9。现在问搬寝室的最小疲劳度为多少。

解题思路

我们先将这n件物品按重量排序，这样每次从中选取相邻的k对物品，那么左右手的重量差最小，因此疲劳度也会比较小。
令dp[i][j]表示前i个物品中选j对的最小疲劳度。则dp[i][j]可能含有第i个物品(这种情况下,第i种物品一定是和第i-1个物品配对)，则dp[i][j]=dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1])；dp[i][j]的j对也可能不含有第i个物品，此时有dp[i][j]=dp[i-1][j]

代码部分

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
using namespace std;
const int inf=0x3f3f3f3f;
int dp[2500][1500],val[2500];
int main()
{
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0; i<n; i++)
        {
            scanf("%d",&val[i]);
        }
        sort(a,a+n);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=k; 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]+(val[i-1]-val[i-2])*(val[i-1]-val[i-2]));
            }
        }
        printf("%d\n",dp[n][k]);
    }
    return 0;
}