AcWing 10. 有依赖的背包问题

树形dp + 分组背包

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

#define fr first
#define se second

typedef pair<int, int> PII;
typedef long long LL;
typedef unsigned long long ULL;

const int INF = 0X3f3f3f3f, N = 100 + 10, M = 3e4+10, MOD = 1e9 + 7;

int n,m;
int v[N],w[N];
int f[N][M];

vector<int> tr[N];

void dfs(int cur){ 
    for(int i=m;i>=v[cur];i--) f[cur][i]=w[cur]; // 子树依赖于当前点，所以肯定要选当前的结点
    
    for(int i=0;i<tr[cur].size();i++){
        int to=tr[cur][i];
        dfs(to); // 先递归求出f[to][0~m]
        
        for(int j=m;j>=v[cur];j--) // 转化为分组背包，先枚举剩余体积，剩余体积小于v[cur]无意义
            for(int k=0;k+v[cur]<=j;k++) // 再枚举子树的体积，子树体积+v[cur]不能超过剩余体积
                f[cur][j]=max(f[cur][j],f[cur][j-k]+f[to][k]);    
    }
}

void work() {
    int root;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
	    int a,b,c;
	    cin>>a>>b>>c;
	    v[i]=a,w[i]=b;
	    if(c!=-1) tr[c].push_back(i);
	    else root=i;
	}
	dfs(root);
	cout<<f[root][m];
}

signed main() {
	
	work();
	
	return 0;
}