P1757 通天之分组背包 题解

通天之分组背包 - 题解

题目背景

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

题目描述

自 0-1 背包问世之后，小 A 对此深感兴趣。一天，小 A 去远游，却发现他的背包不同于 0-1 背包，他的物品大致可分为 $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

数据范围

• $0\le m\le 1000$
• $1\le n\le 1000$
• $1\le k\le 100$
• $a_i,b_i,c_i$ 在 int 范围内。

题目分析

这是一个典型的分组背包问题。分组背包问题是 0-1 背包问题的扩展，区别在于每组中最多选一个物品。

解题思路

1. 初始化和输入处理:

   • 读取总重量 m 和物品数量 n。
   • 根据所属组数，将物品的重量和价值存储到对应组的数组中。
   • 记录最大的组编号 K。

2. 动态规划数组初始化:

   • 定义 dp[j] 表示总重量不超过 j 时的最大价值。

3. 分组背包状态转移:

   • 对每个组进行处理时，从总重量 m 开始逆序更新 dp 数组，以防止同一组内的物品重复使用。
   • 对当前组的每个物品，更新 dp 数组，计算在选择该物品的情况下，最大价值是多少。

代码实现

以下是最终实现的代码：

#include <bits/stdc++.h>
using namespace std;

int n, m;
vector<int> w[114], v[114];
int dp[1145];
int K;

int main() {
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    cin >> m >> n;
    for (int i = 1; i <= n; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        w[c].push_back(a);
        v[c].push_back(b);
        K = max(K, c);
    }
    fill(dp, dp + m + 1, 0);
    for (int k = 1; k <= K; k++) {
        for (int j = m; j >= 0; j--) {
            for (int i = 0; i < w[k].size(); i++) {
                if (j >= w[k][i]) {
                    dp[j] = max(dp[j], dp[j - w[k][i]] + v[k][i]);
                }
            }
        }
    }
    cout << dp[m] << endl;
    return 0;
}

代码解释

初始化: 使用 fill 函数初始化 dp 数组为 0。
状态转移:
- 遍历每个组（从 1 到最大组编号 K）。
- 对每个组，从总重量 m 开始逆序遍历 dp 数组。
- 对于组内每个物品，更新 dp[j]，计算在选择该物品的情况下的最大价值。

关键点

• 逆序遍历: 确保在处理每个组时，不会重复使用同一组的物品。