树形dp＜三＞

有依赖的背包问题

原题连接

有 N 个物品和一个容量是 V 的背包。

物品之间具有依赖关系，且依赖关系组成一棵树的形状。如果选择一个物品，则必须选择它的父节点。

如下图所示：

如果选择物品5，则必须选择物品1和2。这是因为2是5的父节点，1是2的父节点。

每件物品的编号是 i，体积是 vi，价值是 wi，依赖的父节点编号是 pi。物品的下标范围是 1…N。

求解将哪些物品装入背包，可使物品总体积不超过背包容量，且总价值最大。

输出最大价值。

输入格式
第一行有两个整数 N，V，用空格隔开，分别表示物品个数和背包容量。

接下来有 N 行数据，每行数据表示一个物品。
第 i 行有三个整数 vi,wi,pi，用空格隔开，分别表示物品的体积、价值和依赖的物品编号。
如果 pi=−1，表示根节点。 数据保证所有物品构成一棵树。

输出格式
输出一个整数，表示最大价值。

数据范围
1≤N,V≤100
1≤vi,wi≤100
父节点编号范围：

内部结点：1≤pi≤N;
根节点 pi=−1;
输入样例
5 7
2 3 -1
2 2 1
3 5 1
4 7 2
3 6 2
输出样例：
11

题解

本题是一个树形dp + 分组背包问题。
树形dp一般是dfs之后，用子节点的值更新父节点，分组背包就是一组只能选择一种物品或者不选的背包问题。

对于每一个节点来说是一个分组背包问题，其中节点被分成了$[0,j-v[u]]$种不同的组，每一组的体积就是$[0,j-v[u]]$，每一组只能选择一种物品。

1. 对于每一个节点首先进行初始化，如果体积小于了v[u]初始化为0， 否则初始化为w[u]，因为如果可以选择本节点，最小也是本节点的价值。
2. 求出子节点分别有不同体积下的最大值。
3. 使用分组背包问题求解，j范围是[v[u], m]逆序，这是分组背包的01性决定的
4. k代表决策，也就是可以分给子树多少容量的题解，所以是[0, j - v[u]]

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int>PR;

const int N = 105;
int n, m;
int w[N], s[N];
vector<int>G[N];
int f[N][N];

void dfs(int u) {
    for (int i = s[u]; i <= m; i++) f[u][i] = w[u];     // 初始化为当前节点的值
    for (int v: G[u]) {                                 // 种类
        dfs(v);
        for (int j = m; j >= s[u]; j--) {               // 体积
            for (int k = 0; k <= j - s[u]; k++) {       // 决策
                f[u][j] = max(f[u][j], f[u][j - k] + f[v][k]);
            }
        }
    }
}

int main() {
    cin >> n >> m;
    int root;
    for (int i = 1; i <= n; i++) {
        int q;
        cin >> s[i] >> w[i] >> q;
        if (q == -1) root = i;
        else G[q].push_back(i);
    }
    
    dfs(root);

    printf("%d\n", f[root][m]);
    
    return 0;
}

苹果二叉树

原题链接

也是一个有依赖的背包问题，对于每一个节点，是一个分组背包问题，组内的物品是根据选择枝桠的数量划分的。和上一个问题不同的是，权重在边上，而不是在点上。影响就是：

1. 如果访问子节点，需要预留一条边，那么就只剩下了$j-1$的体积,所以转移是$f[j-k-1]$。
2. 另外权重在边上，不能在访问点外进行初始化。

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int>PR;
const int N = 105;
int f[N][N];
vector<PR>G[N];

int n, m;

void dfs(int u, int fa) {
    for (auto &[v, w] :G[u]) {
        if (v == fa) continue;
        dfs(v, u);
        for (int j = m; j >= 1; j--) {
            for (int k = 0; k < j; k++) {
                f[u][j] = max(f[u][j - k - 1] + f[v][k] + w, f[u][j]);
            }
        }
    }
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n - 1; i++) {
        int a, b, w;
        cin >> a >> b >> w;
        G[a].emplace_back(b, w);
        G[b].emplace_back(a, w);
    }

    dfs(1, -1);

    printf("%d\n", f[1][m]);

    return 0;
}