HDU 1421 搬寝室

搬寝室

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

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

 

#include<cstdio>
#include<algorithm>
using namespace std;

int dp[2000][1000];  //dp[i][j]表示前i件物品中选择j对物品所获得的最小疲劳度

int main()
{
    int n,k,min,i,j;
    int w[2000];
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        for(i=1;i<=n;++i)
            scanf("%d",&w[i]);
        sort(w+1,w+n+1);     //先排序，因为选择的一对物品肯定相邻才能保证差值最小从而疲劳度最小
        memset(dp,0,sizeof(dp));
        dp[2][1]=(w[1]-w[2])*(w[1]-w[2]);
        for(i=3;i<=n;++i)
            for(j=0;j<=i/2&&j<=k;++j)
                if(i-1>=2*j)  //dp[i][j]=min(前i-1件物品中选择j对所获得的最小疲劳度,前i-2件物品中选择j-1对再加上第i-1件和第i件所获得的最小疲劳度)
                    dp[i][j]=dp[i-1][j]<dp[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1])?dp[i-1][j]:dp[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]);
                else         // 前i-1件中不能跳出j对物品所以只能把第i件选上
                    dp[i][j]=dp[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]);
        /*
        min=0x7fffffff;     //多此一举
        for(i=2*k;i<=n;++i)
            if(dp[i][k]<min)
                min=dp[i][k];
        */
        printf("%d\n",dp[n][k]);   //dp[n][k] 已经表示了n件物品中选择k对获得的最小疲劳度
    }
    return 0;
}