AcWing 10. 有依赖的背包问题（树形dp + 分组背包）

题意

每选择一个物品就应当选择其父节点，问所有选择方案能得到的最大价值是多少？

思路

一道 背包DP 的 变种题目

根据题设的 拓扑结构 可以观察出每个 物品 的关系构成了一棵 树

而以往的 背包DP 每个 物品 关系是 任意的（但我们一般视为 线性的）

所以，这题沿用 背包DP 的话，要从原来的 线性DP 改成 树形DP 即可

然后思考 树形DP 的 状态转移

先比较一下以往 线性背包DP 的 状态转移，第 i 件 物品 只会依赖第 i−1 件 物品 的状态

如果本题我们也采用该种 状态依赖关系 的话，对于节点 i，我们需要枚举他所有子节点的组合 2^k 种可能

再枚举 体积，最坏时间复杂度 可能会达到 O(N × 2^N × V)（所有子节点都依赖根节点），最终毫无疑问会 TLE

因此我们需要换一种思考方式，那就是枚举每个 状态 分给各个子节点 的 体积

这样 时间复杂度 就是 O(N × V × V)

具体分析如下：

设节点 u 有 i 个子节点 s1，s2，s3，...，si，每个子节点可选或不选。

我们可以把 u 的 i 个子节点看做 i 组物品，每组物品 si 按单位体积拆分，有 0，1，2，…，j - v[u] 种决策。

按单位体积拆分是因为 si 的子孙可能存在体积为 1 的物品。

拆分到 j - v[u] 是因为要预留出 v[u] 的空间装入节点 u 的物品。

从根节点 u 开始 dfs，从根到叶，再从叶到根回溯时，做**分组背包**，更新 f 值。

目标状态 f[root][m]：root是根节点，表示选择 以整棵树的根节点 u 为子树的物品，体积和不超过最大容量 m 所获得的最大价值（答案）

时间复杂度:

O(N × V × V)

代码：

#include<bits/stdc++.h>

using namespace std;

int n, m;
const int N = 110;
vector<int> g[N];
int dp[N][N], v[N], w[N];

void dfs(int u)
{
    for(int i=v[u]; i<=m; ++i) dp[u][i] = w[u]; //点 u 必须选，所以初始化 dp[u][v[u] ~ m] = w[u]
    for(auto son : g[u]) //物品
    {
        dfs(son); //递归儿子 son 每个儿子是一组物品
        for(int j=m; j>=v[u]; --j) //体积：j 的范围为 v[u]~m，小于 v[u] 无法选择以 u 为子树的物品
            for(int k=0; k<=j-v[u]; ++k) //决策：分给子树 son 的空间不能大于 j - v[u]，不然无法选根物品 u
            dp[u][j] = max(dp[u][j], dp[u][j-k]+dp[son][k]); //每种决策对应一个打包的物品，可能包含多个的子孙物品的组合。
    }
}

int main()
{
    cin>>n>>m;
    int root = 0; //存储整棵树的根节点编号
    for(int i=1; i<=n; ++i)
    {
        int p;
        cin>>v[i]>>w[i]>>p;
        if(p==-1) root = i; //找到整棵树的根节点
        else g[p].push_back(i);
    }
    dfs(root); //从根节点往下递归处理
    cout<<dp[root][m]<<endl;
    
    return 0;
}