洛谷P1858 多人背包【0/1背包前k优解】

题目描述

求01背包前k优解的价值和

DD 和好朋友们要去爬山啦！

他们一共有 K 个人，每个人都会背一个包。这些包 的容量是相同的，都是 V。可以装进背包里的一共有 N 种物品，每种物品都有 给定的体积和价值。

在 DD 看来，合理的背包安排方案是这样的： 每个人背包里装的物品的总体积恰等于包的容量。 每个包里的每种物品最多只有一件，但两个不同的包中可以存在相同的物品。

任意两个人，他们包里的物品清单不能完全相同。 在满足以上要求的前提下，所有包里的所有物品的总价值最大是多少呢？

对于100%的数据，K≤50,V≤5000,N≤200

题目分析

简化题意：求01背包前k优解的价值和，背包必须装满
设$dp[j][k]$表示总体积为j时的第k优解

为了方便理解，讲解时先把第一维加回来
易知$dp[i][j]$的前k优解一定是$dp[i-1][j][1...k]$和$dp[i-1][j-w_i][1...k]+v_i$这2k个数中的k个
而$dp[i-1][j][1...k]$和$dp[i-1][j-w_i][1...k]+v_i$又各自有序
所以只要用类似归并排序里每一层合并两有序数组的方法就能求出其中的前k大值

把第一维去掉按0/1背包的方法枚举即可

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

int k,n,m;
int w[210],val[210];
int dp[5010][55];
int rem[55],ans;

int main()
{
    k=read();m=read();n=read();
    for(int i=1;i<=n;++i)
    w[i]=read(),val[i]=read();
    
    memset(dp,128,sizeof(dp)); dp[0][1]=0;
    for(int i=1;i<=n;++i)
    {
        for(int j=m;j>=w[i];--j)
        {
            int t1=1,t2=1,len=0;
            while(t1+t2<=k+1)
            {
                if(dp[j][t1]>dp[j-w[i]][t2]+val[i]) rem[++len]=dp[j][t1++];
                else rem[++len]=dp[j-w[i]][t2++]+val[i];
            }
            for(int l=1;l<=k;++l) dp[j][l]=rem[l];
        }
    }
    for(int i=1;i<=k;++i) ans+=dp[m][i];
    printf("%d\n",ans);
    return 0;
}