AcWing 10. 有依赖的背包问题 （树形DP）

题目链接：10. 有依赖的背包问题 - AcWing题库

---

题目大意：

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

每个物品有体积 w[i]，价值 v[i]

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

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define  mst(a,x) memset(a,x,sizeof (a))
using namespace std;
typedef pair<int, int> pii;

const int N = 1008, mod = 998244353;

int n, m;
int h[N], ne[N], e[N], idx;
int w[N], v[N];
int f[N][N];
int root;

void add(int a, int b) {
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

void dfs(int u) {

    for (int i = h[u]; ~i; i = ne[i]) {
        int son = e[i];
        dfs(son);
        
        for (int j = m; j >= w[son]; j--) {
            for (int k = w[son]; k <= j; k++) {
                f[u][j] = max(f[u][j], f[son][k] + f[u][j - k]);//先尽可能选儿子节点
            }
        }
    }
    for (int i = m; i >= w[u]; i--) f[u][i] = f[u][i - w[u]] + v[u];//一定选自己
    for (int i = 0; i < w[u]; i++) f[u][i] = 0;//无法选该节点，也就无法选儿子节点，价值为0
}

void solve() {

    mst(h, -1);
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        int c;
        cin >> w[i] >> v[i] >> c;
        if (c == -1) root = i;
        else add(c, i);
    }
    
    dfs(root);
    
    cout << f[root][m] << endl;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
// cin >> T;
    while (T--) solve();
    
    return 0;
}