分组背包 模板 C++实现

问题：分组背包就是，给你 n 组物品，然后每一组你至多选择一个物品(也可以不选),每个物品都有自己的体重量 weight 和价值 value ，现在给你一个容量为 c 的背包，让你用这个背包装物品，使得物品价值总和最大。

weight 重量数组，value 价值数组 ，s 每组元素个数数组，n 组数，c 背包容量

---

算法1：朴素写法（三重循环）

dp [ i ][ j ] 一定大于等于 dp [ i - 1][ j ] ，所以先把 dp [ i - 1][ j ] 复制过来，然后遍历每一组。

代码：

for (int i = 1; i <= n; i++)
    for (int j = 0; j <= c; j++){
        dp[i][j] = dp[i - 1][j];// 把上一行的复制过来
        for (int k = 1; k <= s[i - 1]; k++)
            if (weight[i - 1][k - 1] <= j)  dp[i][j] = max(dp[i][j], dp[i - 1][j - weight[i - 1][k - 1]] + value[i - 1][k - 1]);
    }

---

算法2：

将其转换为一维数组，但是第二层循环要逆置。因为每个物品只能选择一次。

代码：

for (int i = 1; i <= n; i++)
    for (int j = c; j >= 0; j--)
        for (int k = 1; k <= s[i - 1]; k++)
            if (weight[i - 1][k - 1] <= j)  dp[j] = max(dp[j], dp[j - weight[i - 1][k - 1]] + value[i - 1][k - 1]);

完整代码：

#include<iostream>

using namespace std;

const int N = 10;
int main(){
    int n = 3, c = 5;// 组数n 背包容量c
    int weight[N][N] = { {1,2},{ 3 }, { 4 } };
    int value[N][N] = { {2,4}, { 4 }, { 5 } };
    int s[N] = { 2,1,1 };// 每组元素个数
/*
    // 方法1 朴素
    int dp[N][N];
    memset(dp, 0, sizeof(dp));// 初始化dp
    for (int i = 1; i <= n; i++)
        for (int j = 0; j <= c; j++){
            dp[i][j] = dp[i - 1][j];// 把上一行的复制过来
            for (int k = 1; k <= s[i - 1]; k++)
                if (weight[i - 1][k - 1] <= j)  dp[i][j] = max(dp[i][j], dp[i - 1][j - weight[i - 1][k - 1]] + value[i - 1][k - 1]);
        }
    cout << dp[n][c] << endl;
*/

    // 方法2 
    int dp[N] = { 0 };
    for (int i = 1; i <= n; i++)
        for (int j = c; j >= 0; j--)
            for (int k = 1; k <= s[i - 1]; k++)
                if (weight[i - 1][k - 1] <= j)  dp[j] = max(dp[j], dp[j - weight[i - 1][k - 1]] + value[i - 1][k - 1]);
    cout << dp[c] << endl;

    return 0;
}