hdoj1421-搬寝室

hdoj1421-搬寝室

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

思路

      本题采用动态规划，用数组dp[i][j]记录从i件物品里搬j件所需疲劳值的最小值。
      首先将数据排序，这样相邻的两个数的差平方值最小，然后对于第i件物品；

• 当不选第i件物品时，dp[i][j] = dp[i-1][j]
• 当选择第i件物品时，dp[i][j] = dp[i-2][j-1] + (data[i]-data[i-1])^2
• 由上，得到状态转移方程为
  dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(data[i]-data[i-1])*(data[i]-data[i-1]))

code

//Exe.Time  Exe.Memory  
// 780MS     17544K
#include <iostream>
#include <fstream>
#include <cstring>
#include <cmath>
#include <algorithm> 

using namespace std;

int data[2005];
int dp[2005][2005];
int sq(int n)
{
    return n * n;
}

int main()
{
    //ifstream cin("data.in");
    int n, k;
    while(cin >> n >> k)
    {
        memset(dp, 0x3f3f3f3f, sizeof(dp));
        for(int i = 1; i <= n; i ++)
        {
            cin >> data[i];
        }
        dp[0][0] = dp[1][0] = 0;
        sort(data+1, data+n+1);
        for(int i = 2; i <= n; i ++)
        {
            dp[i][0] = 0;
            int k = i / 2;
            for(int j = 1; j <= k; j ++)
            {
                //状态转移方程
                dp[i][j] = min(dp[i-1][j], dp[i-2][j-1] + sq(data[i] - data[i-1]));
            }
        }
        cout << dp[n][k] << endl;
    }
    return 0;
}