hdu1421搬寝室(动态规划)

搬寝室

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 33261    Accepted Submission(s): 11387

Problem Description

搬寝室是很累的,xhd深有体会.时间追述2006年7月9号,那天xhd迫于无奈要从27号楼搬到3号楼,因为10号要封楼了.看着寝室里的n件物品,xhd开始发呆,因为n是一个小于2000的整数,实在是太多了,于是xhd决定随便搬2*k件过去就行了.但还是会很累,因为2*k也不小是一个不大于n的整数.幸运的是xhd根据多年的搬东西的经验发现每搬一次的疲劳度是和左右手的物品的重量差的平方成正比(这里补充一句,xhd每次搬两件东西,左手一件右手一件).例如xhd左手拿重量为3的物品,右手拿重量为6的物品,则他搬完这次的疲劳度为(6-3)^2 = 9.现在可怜的xhd希望知道搬完这2*k件物品后的最佳状态是怎样的(也就是最低的疲劳度),请告诉他吧.

 

Input

每组输入数据有两行,第一行有两个数n,k(2<=2*k<=n<2000).第二行有n个整数分别表示n件物品的重量(重量是一个小于2^15的正整数).

 

Output

对应每组输入数据,输出数据只有一个表示他的最少的疲劳度,每个一行.

 

Sample Input

2 1

1 3

 

Sample Output

4

 

题意：有n件物品，需要搬2*k件，每次般2件，左手一件，右手一件，搬物品需要消耗疲劳度（左手的重量与右手的重量之差的平方）,需要输出一个最少的疲劳度

题解：先把物品按照重量从小到大排，这样肯定是相邻的两个一起拿比较省力，动态规划的时候分2种情况，拿和不拿，拿的话就是dp[i-1][j-2]+(a[j]-a[j-1])*(a[j]-a[j-1])。不拿就直接是dp[i][j-1].。所以状态转移方程为dp[i][j]=min(dp[i][j-1],dp[i-1][j-2]+(a[j]-a[j-1])*(a[j]-a[j-1]))

#include<bits/stdc++.h>
using namespace std;
int a[2005],dp[2005][2005];
int main() {
    int n,k;
    while(~scanf("%d %d",&n,&k)) {
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++) {
            scanf("%d",&a[i]);
        }
        sort(a+1,a+1+n);
        for(int i=1; i<=k; i++)//2k件也就是k对 
            for(int j=1; j<=n; j++)
                if(j>=2*i) {
                    dp[i][j]=min(dp[i][j-1],dp[i-1][j-2]+(a[j]-a[j-1])*(a[j]-a[j-1]));
                
                }
                else
                    dp[i][j]=1e9;

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