背包问题——多重背包

问题描述

有N种物品和一个容量为W的背包。第i种物品最多有n[i]件可用，每件重量是w[i]，价值是v[i]。求解将哪些物品装入背包可使这些物品的重量总和不超过背包容量，且价值总和最大。

分析：

两个最基本的算法：

1.使用三重循环进行递推：
状态转移式为：dp[i][j]=max{dp[i-1][j-k*w[i]]+k*v[i]|0<=k<=n[i]}

关键代码如下：

void solve()
{
    for(int i=0; i<N; i++)
    {
        for(int j=0; j<=W; j++)
        {
            for(int k=0; k<=a[i] && k*w[i]<=j; k++)
            {
                dp[i+1][j] = max(dp[i][j], dp[i][j-k*w[i]]+k*v[i]); 
            }
        }
    }
    printf("%d\n",dp[N][W]);
}

然而复杂度为O(V*Σn[i])，效率太低。

2.转换为01背包
即把第i种物品换成n[i]件01背包中的物品，这样就转换为求Σn[i]的01背包问题，复杂度仍为O(V*Σn[i])。
代码如下：

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100+10;
const int max_n = 1000000;
int w[maxn], v[maxn], a[maxn];
int w_n[max_n], v_n[max_n];
int dp[max_n];
int N, W;
int k; 
/*01背包*/
void solve()
{
    for(int i=0; i<k; i++)
    {
        for(int j=W; j>=w_n[i]; j--)
        {
            dp[j] = max(dp[j], dp[j-w_n[i]]+v_n[i]);    
        }   
    }
    printf("%d\n",dp[W]);
}

int main()
{
    scanf("%d%d",&N,&W);
    k=0;
    for(int i=0; i<N; i++)
    {
        scanf("%d%d%d",&w[i],&v[i],&a[i]);
        //将第i件物品转换为n[i]件01物品
        for(int j=0; j<a[i]; j++)
        {
            w_n[k] = w[i];
            v_n[k] = v[i];
            k++;
        }
    }
    solve();
    return 0;
}

---

最优算法：

将第i种物品分成若干件物品，其中每件物品有一个系数，这件物品的重量和价值均是原来的重量和价值乘以这个系数。使这些系数分别为1,2,4,…,2^(k-1),n[i]-2^k+1，且k是满足n[i]-2^k+1>0的最大整数。例如，如果n[i]为13，就将这种物品分成系数分别为1,2,4,6的四件物品。
这样就将第i种物品分成了O(log n[i])种物品，将原问题转化为了复杂度为O(V*Σlog n[i])的01背包问题。
——摘自《背包九讲》

这就是二进制法。

代码如下：

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn = 100+10;
int N,W; 
int w[maxn], v[maxn], a[maxn]; 
int dp[maxn];
/*01背包*/
void zero_pack(int w, int v)
{
    for(int j=W; j>=w; j--)
    {
        dp[j] = max(dp[j], dp[j-w]+v);
    }
}
/*完全背包*/
void comp_pack(int w, int v)
{
    for(int j=w; j<=W; j++)
    {
        dp[j] = max(dp[j], dp[j-w]+v);
    }
}
/*多重背包*/
void mult_pack(int w, int v, int a)
{
    //如果某件物品的数量乘以重量大于等于W，那么这个物品的最优解就直接用完全背包来推。
    if(w * a >= W)
    {
        comp_pack(w, v);
        return;
    }
    int k = 1;
    while(k < a)
    {
        zero_pack(k*w, k*v);
        a = a - k;
        k = k*2;
    }
    //剩余的再需要一次01背包
    zero_pack(a*w, a*v);
}

int main()
{
    scanf("%d%d",&N, &W);
    for(int i=0; i<N; i++)
    {
        scanf("%d%d%d",&w[i],&v[i],&a[i]);  
    } 
    for(int i=0; i<N; i++)
    {
        mult_pack(w[i], v[i], a[i]);
    }
    printf("%d\n",dp[W]);
    return 0;
}