HDU - 1421 搬寝室 - DP

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

动态规划：

定义数组a[i]为搬第i对物品所消耗的疲劳值；数组f[n][k]来表示在n件物品中搬k对的最佳状态，而达到这一状态的决策可能为：

1.第n件物品不搬，即在前n - 1件物品中搬k对，那么疲劳值仍为f[n - 1][k]；

2.第n件物品要搬，那么根据上面所证，第n - 1件物品也要同时搬，即在前n - 2件物品中搬k - 1对物品，再搬最后一对物品，那么疲劳值为f[n - 2][k - 1] + a[n - 1]，n - 1是因为对数必然比总物品数少1。

dp[i][j]=min(dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]),dp[i-1][j])  

 

#include <stdio.h>
#include <string.h>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include<queue>
#include<stack>
using namespace std;
int a[2010];
int dp[2010][2010];

int main()
{
    int k,n;
    dp[0][0]=0;
    
    while(~scanf("%d%d",&n,&k))
    {
        memset(a,0,sizeof(a));
        int i,j;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);

        sort(a+1,a+n+1);
//        for(i=1;i<=n;i++)
//          printf("%d**",a[i]);
        for(i=0;i<=n;i++)
           for(j=1;j<=k;j++)
              dp[i][j]=100000000; // 赋初值-100，MD,坑死 
              
        for(i=2;i<=n;i++)
            for(j=1;2*j<=i;j++)
              dp[i][j]=min(dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]),dp[i-1][j]);
                           
        printf("%d\n",dp[n][k]);
            
    }
    
    return 0;
}