P1757 通天之分组背包（dp）

P1757 通天之分组背包

题目背景

直达通天路·小 A 历险记第二篇

题目描述

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

输入格式

两个数 m,n，表示一共有 n 件物品，总重量为 m。
接下来 n 行，每行 3 个数 ai,bi,ci表示物品的重量，利用价值，所属组数。

输出格式

一个数，最大的利用价值。

输入输出样例

输入 #1

45 3
10 10 1
10 5 1
50 400 2

输出 #1

10

说明/提示

1≤m,n≤1000。

我的思路

分组背包其实和01背包的解法类似，不同的是，分组背包是分组的01背包原理相同

AC代码

#include<bits/stdc++.h>
using namespace std;
int dp[1001], a[1001], b[1001], c[1001], num[1001], bh[1001][1001], m, n, cnt;//bh[1001][1001]用来记录第几组第几个 
int main() {
  cin >> m >> n;
  for (int i = 1; i <= n; i++) {
    cin >> a[i] >> b[i] >> c[i];
    cnt = max(cnt, c[i]);//求最大组数 
    num[c[i]]++;//记录改组有多少个 
    bh[c[i]][num[c[i]]] = i;
  }
  for (int i = 1; i <= cnt; i++) 
    for (int j = m; j >= 0; j--) 
      for (int k = 1; k <= num[i]; k++) //从每组中取最优解 
        if (j >= a[bh[i][k]])
          dp[j] = max(dp[j], dp[j - a[bh[i][k]]] + b[bh[i][k]]);                    
  cout << dp[m];
  return 0;
}