hdu 2639 Bone Collector II （背包的第K优解）

Bone Collector II

原题链接： http://acm.hdu.edu.cn/showproblem.php?pid=2639

题意：普通的背包问题，主要是求第k优解

思路：我们就把普通的背包新增一维dp[i][j]记录背包容量为i的时候第j大的价值为dp[i][j],对于每一件物品b数组记录取了这件物品之后的前k大价值，a数组记录不取这件物品的前k大价值，然后把ab数组合并去重取前k个记录下来就可以了，这样可以保证答案的正确性。

代码：

#include<stdio.h>
#include<string.h>
int n,m,k;
int dp[1010][55],a[1010],b[1010],v[1010],w[1010];
void zonepack()
{
    int i,j,ans,x,y,z;
    memset(dp,0,sizeof(dp));
    for(i=1;i<=n;i++)
    {
        for(j=m;j>=w[i];j--)
        {
            for(ans=1;ans<=k;ans++)
            {
                a[ans]=dp[j][ans];//ab模拟取第i件物品和不取的过程，因为算法过程a[1~t]是一个单调递减的数列,b[1~t]也一样
                b[ans]=dp[j-w[i]][ans]+v[i];
            }
            a[ans]=-1,b[ans]=-1;
            x=y=z=1;
            while(z<=k&&(a[x]!=-1||b[y]!=-1))//对ab进行新的排序,每次只取前k个
            {
                if(a[x]>b[y])//排序
                {
                    dp[j][z]=a[x];
                    x++;
                }
                else
                {
                    dp[j][z]=b[y];
                    y++;
                }
                if(dp[j][z]!=dp[j][z-1])//去重
                    z++;
            }
        }
    }
    printf("%d\n",dp[m][k]);
}
int main()
{
    int t,i;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=n;i++)
            scanf("%d",&v[i]);
        for(i=1;i<=n;i++)
            scanf("%d",&w[i]);
        zonepack();
    }
    return 0;
}