AcWing 1074 二叉苹果树 题解 （动态规划—DP—树形DP）

原题传送门

#include<bits/stdc++.h>

using namespace std;

const int N = 110, M = 2 * N;

int n, m;
int h[N], e[M], ne[M], w[M], idx;
int f[N][N];//f[i][j]表示以i为根节点的子树，保留j条树枝能留住的最大苹果数量 

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

void dfs(int u, int father){
	for(int i = h[u]; i != -1; i = ne[i]){
		if(e[i] == father) continue;
		dfs(e[i], u);
		/*分组背包，将每个子节点视为一个物品*/
		for(int j = m; j >= 0; j -- ){//枚举体积 
			for(int k = 0; k + 1 <= j; k ++ ){//枚举决策，就是保留多少树枝，因为这个节点向父节点连还需要一个树枝，所以k + 1 <= j 
				f[u][j] =max(f[u][j], f[u][j - k - 1] + f[e[i]][k] + w[i]);
			}
		}
	}
}

int main()
{
	cin>>n>>m;
	memset(h, -1, sizeof h);
	for(int i = 1; i < n; i ++ ){
		int a, b, c;
		cin>>a>>b>>c;
		add(a, b, c);
		add(b, a, c);
	}
	
	dfs(1, -1);
	
	cout<<f[1][m]<<endl;
	return 0; 
}