HDU 2639(Bone Collector II)背包问题求第K优解

背包问题求第K优解，此类题在背包九讲里有说到过，刚好也是以01背包为例的：

其基本思想是，将每个状态都表示成有序队列，将状态转移方程中的max/min 转化成有序队列的合并。
这里仍然以01 背包为例讲解一下。
首先看01背包求最优解的状态转移方程：f[i][v]=max{f[i-1][v],f[i-1][v-c[i]]+w[i]}。如果要求第K优解，那么状态f[i][v]就应该是一个大小为K的数组f[i][v][1..K]。其中f[i][v][k]表示前i个物品、背包大小为 v时，第k优解的值。“f[i][v]是一个大小为K的数组”这一句，熟悉C语言的同学可能比较好理解，或者也可以简单地理解为在原来的方程中加了一维。 显然f[i][v][1..K]这K个数是由大到小排列的，所以我们把它认为是一个有序队列。然后原方程就可以解释为：f[i][v]这个有序队列是由f[i-1][v]和f[i-1][v-c[i]]+w[i]这两个有序队列合并得到的。有序队列f[i-1][v]即f[i-1][v][1..K]，f[i-1][v-c[i]]+w[i]则理解为在f[i-1][v-c[i]] [1..K]的每个数上加上w[i]后得到的有序队列。合并这两个有序队列并将结果的前K项储存到f[i][v][1..K]中的复杂度是O(K)。最后的答案是f[N][V][K]。总的复杂度是O(VNK).

当然可以用空间优化过的01背包算法。看到网上有个比喻这种方法，2个班级要选出全年级前10的，那就分别选出两个班级的前10的，然后比较下。

此题就是选出每个背包状态的前K优解，运用上述说的两两归并的方法去掉重复的。

代码：

#include<stdio.h>
#include<string.h>

int main()
{
    int T;
    int i,j,k,n,v,knum,ca,cb,cc;
    int val[110],vg[110],dp[1010][110];
    int a[110],b[110];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&v,&knum);
        memset(val,0,sizeof(val));
        memset(vg,0,sizeof(vg));
        for(i=0;i<n;i++)
            scanf("%d",&val[i]);
        for(i=0;i<n;i++)
            scanf("%d",&vg[i]);
        memset(dp,0,sizeof(dp));
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(i=0;i<n;i++)
        {
            for(j=v;j>=vg[i];j--)
            {
                for(k=1;k<=knum;k++)
                {
                    a[k]=dp[j-vg[i]][k]+val[i];
                    b[k]=dp[j][k];
                }//列出前K个可能，不要MAX，MAX的话就会少掉很多情况 
                ca=1,cb=1,cc=1;
                while(cc<=knum && ca<=knum || cb<=knum)  //把这几种可能合并下，并排序（过程如同两两归并排序），去掉重复的 
                {
                    if(a[ca]>b[cb])
                    {
                        dp[j][cc]=a[ca];
                        ca++;
                    }
                    else 
                    {
                        dp[j][cc]=b[cb];
                        cb++;
                    } 
                    if(dp[j][cc]!=dp[j][cc-1])
                        cc++;
                }
            }        
        }
        printf("%d\n",dp[v][knum]);
    }
    return 0;
}