HDU 2639 Bone Collector II(DP:01背包)

HDU 2639 Bone Collector II(DP:01背包)

http://acm.hdu.edu.cn/showproblem.php?pid=2639

题意:同样有N个物品,每个物品有weight和value,问你在总重量不超过W的情况下,使得value值尽量大.得到的第K大的value值是多少?

分析:

       首先我们令dp[i][Wx][k]=y表示决策完前i个物品,背包容量为Wx时能获得的第k大value值为y.

由动态规划问题的无后效性和阶段性决策我们可以知道,当我们比如求完选择前i个物品后获得了前K大的容量为Wx的背包的最大value值为dp[i][Wx][1], dp[i][Wx][2],…dp[i][Wx][k]后,我们要计算第i+1个物品选还是不选的时候:

首先dp[i+1][Wx][1], dp[i+1][Wx][2],…dp[i+1][Wx][k],的这k个值一定是由dp[i][Wx][1], dp[i][Wx][2],…dp[i][Wx][k]或

dp[i][Wx-weight[i]][1]+value[i],

dp[i][Wx-weight[i]][2]+value[i],

…,

dp[i][Wx-weight[i]][k]+value[i],

这2K个值取前k个组合而成的.且其中dp[i][Wx][1]是第一个队列里面最大的,而dp[i][Wx-weight[i]][1]+value[i]是第二个队列里面最大的,现在就等于两个有序链表合并去前K个一样了.

上面两个序列可以分别用A数组和B数组保存,然后合并取前K大值即可.

AC代码:注意细节

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXW=1005;
const int MAXN=1005;
const int MAXK=35;
int dp[MAXW][MAXK],weight[MAXN],value[MAXN];
int nKind,nWeight;
int K,A[MAXK],B[MAXK];
void ZeroOnePack()
{
    for(int i=1;i<=nKind;i++)
        for(int j=nWeight;j>=weight[i];j--)
        {
            for(int k=1;k<=K;k++)
            {
                A[k]=dp[j][k];
                B[k]=dp[j-weight[i]][k]+value[i];
            }
            int a=1,b=1,c=1;
            while( (a<=K||b<=K) && c<=K)
            {
                if(b>K ||(a<=K && A[a]>=B[b]) )
                    dp[j][c]=A[a++];
                else
                    dp[j][c]=B[b++];
                if(dp[j][c]!=dp[j][c-1])c++;//注意
            }
            while(c<=K)dp[j][c++]=0;//注意
        }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&nKind,&nWeight,&K);
        for(int i=1;i<=nKind;i++)
            scanf("%d",&value[i]);
        for(int i=1;i<=nKind;i++)
            scanf("%d",&weight[i]);
        memset(dp,0,sizeof(dp));
        ZeroOnePack();
        printf("%d\n",dp[nWeight][K]);
    }
    return 0;
}