DP：搬寝室

F - 搬寝室

Time Limit:1000MS    Memory Limit:32768KB    64bit IO Format:%I64d & %I64u

Submit Status

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 <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

int f[2010][1005];            //这道题用long long 就超时了，很神奇
int v[2010], vv[2010];

int getF(int x, int y){        //处理边界问题的时候可以调用函数来实现，比较方便
    if(y * 2 > x)
        return 1000000000;
    else if(y == 0)
        return 0;
    else
        return f[x][y];
}

int main(){
    int n, k, i, j;
    while(~scanf("%d %d",&n, &k)){
        for(i = 1; i <= n; i ++){
            scanf("%d", &v[i]);
        }
        sort(v + 1, v + n + 1);
        for(i = 1; i <= n-1; i ++){
            v[i] = v[i+1] - v[i];
            v[i] *= v[i];                //预先算出来，节省时间
        }
        memset(f, 0, sizeof(f));
        for(j = 0; j <= k; j ++){  
            for(i = j * 2; i <= n; i ++){
                f[i][j] = min(getF(i-1, j), getF(i-2, j-1) + v[i-1]);
            }
        }
        printf("%d\n",f[n][k]);
    }
    return 0;
}

也可以这样，for循环的顺序改一下：

        for(i = 2; i <= n; i ++){
            for(j = 0; j <= i/2; j ++){
                f[i][j] = min(getF(i-1, j), getF(i-2, j-1) + v[i-1]);
            }
        }