P1757 通天之分组背包（分组背包问题）

题目来自洛谷->P1757 通天之分组背包
题目背景
直达通天路·小 A 历险记第二篇

题目描述
自 01 背包问世之后，小 A 对此深感兴趣。一天，小 A 去远游，却发现他的背包不同于 01 背包，他的物品大致可分为 k 组，每组中的物品相互冲突，现在，他想知道最大的利用价值是多少。

输入格式
两个数 m,n，表示一共有 n 件物品，总重量为 m。

接下来 n行，每行 3个数 a i,b i,c i
​
，表示物品的重量，利用价值，所属组数。

输出格式
一个数，最大的利用价值。
输入输出样例
输入

45 3
10 10 1
10 5 1
50 400 2

输出

10

说明/提示
1≤m,n≤1000。
思路
分组背包与01背包的区别就在于有分组的限制，同一组的物品最多只能选一个。其他的跟01背包大相径庭。
输入时记录每个组中有多少物品（c[zu][0]++;），且记录每个物品的下标（c[zu][c[zu][0]] = i;）。并且更新组号求得最大组号（ k = max(k,zu)）

for (int i = 1;i <= n;i++) {
    cin >> w[i] >> q[i] >> zu;
    c[zu][0]++;//记录每组有多少个物品
    c[zu][c[zu][0]] = i;//记录改组物品下标 
    k = max(k,zu);//找出最大组号 
  }

循环方面跟01背包差不多，就是多个组循环。采用滚动数组优化空间

for (int i = 1;i <= k;i++) //组遍历
    for (int j = m;j > 0;j--) //背包重量逆序遍历
      for (int z = 1;z <= c[i][0];z++) {
        if (j >= w[c[i][z]])
        dp[j] = max(dp[j],dp[j - w[c[i][z]]] + q[c[i][z]]);
      }

直接上代码

#include <bits/stdc++.h>
const int MAx = 1e3 + 1;
using namespace std;
int m,n;
int w[MAx],q[MAx],c[MAx][MAx],zu,k;
int dp[MAx]; 
int main() {
  cin >> m >> n;
  for (int i = 1;i <= n;i++) {
    cin >> w[i] >> q[i] >> zu;
    c[zu][0]++;//记录每组有多少个物品
    c[zu][c[zu][0]] = i;//记录改组物品下标 
    k = max(k,zu);//找出最大组号 
  }
  for (int i = 1;i <= k;i++) //组遍历
    for (int j = m;j > 0;j--)  //背包重量逆序遍历
      for (int z = 1;z <= c[i][0];z++) {//组内遍历
        if (j >= w[c[i][z]])
        dp[j] = max(dp[j],dp[j - w[c[i][z]]] + q[c[i][z]]);
      }
  cout << dp[m];
  return 0;
}