[HDU 1421]搬寝室(富有新意的DP)

标签：style   blog   http   io   os   ar   for   sp   div   

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

题目大意：给出n个物品，要你选出k对物品，使得每对物品之差的平方之和最小。

思路；动态规划求解，首先将物品按重量升序排序，根据贪心思想，每对物品必然是相邻的两个物品，这样重量差的平方才能保证最小。然后DP，用f[i][j]表示前i个物品，选j对的最少代价，这里需要进行一个重要的分类讨论：

(1)如果j*2=i,即前i个物品全部用于配对了，很明显第j对物品是第i-1号物品和第i号物品，f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])^2

(2)其他情况。前i个物品并没有完全用于配对，可以选择第i个物品不配对，这时f[i][j]=f[i-1][j]，或者第i个物品参与配对，同(1)，第i个物品和第i-1个物品配对，f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])^2

最终的答案就是f[n][k]

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 2010

using namespace std;

int f[MAXN][MAXN]; //f[i][j]=前i个物品，选j对的最小花费
int w[MAXN];

int min(int a,int b)
{
    if(a<b) return a;
    return b;
}

int main()
{
    int n,k;
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        memset(f,0,sizeof(f));
        w[0]=0;
        for(int i=1;i<=n;i++)
            scanf("%d",&w[i]);
        sort(w+1,w+n+1);
        for(int i=2;i<=n;i++) //前i个物品
            for(int j=1;j*2<=i;j++) //选j对
            {
                if(i!=2*j) f[i][j]=min(f[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]),f[i-1][j]); //要么第i个物品和第i-1个物品配对，要么留着第i个物品不配对
                else f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]);
            }
        printf("%d\n",f[n][k]);
    }
    return 0;
}

[HDU 1421]搬寝室(富有新意的DP)

标签：style   blog   http   io   os   ar   for   sp   div   

原文地址：http://blog.csdn.net/qpswwww/article/details/40156829