洛谷 P2015 二叉苹果树 【树形DP】

题目链接
题意：

给出一颗树，每个节点有一个正整数权值，你只能保留$m$个节点（节点之间要连通且必须有根节点1），求最大的权值

题解：

树形DP模板题

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
const int mod = 998244353;
using namespace std;
void inar(int a[],int n){for(int i=1;i<=n;i++)cin>>a[i];}
void outar(int a[],int n){for(int i=1;i<=n;i++)cout<<a[i]<<" ";cout<<endl;}
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}

#define endl '\n'
const int INF = 0x3f3f3f3f;
const int N = 1e2+10;
const int maxn = 1e3+10;
struct node{
	int to,w,next;
}e[N];
int head[N],k;
int sz[N],dp[N][N];
int n,q;
void add(int u,int v,int w){
	e[++k].to=v;
	e[k].w=w;
	e[k].next=head[u];
	head[u]=k;
}
void dfs(int u,int fa){
	sz[u]=1;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to,w=e[i].w;
		if(v==fa)continue;
		dfs(v,u);
		sz[u]+=sz[v];
		for(int j=min(q,sz[u]);j>0;j--)//j不能为0，因为父节点必须保留
			for(int k=min(sz[v],j-1);k>=0;k--)//枚举子节点保留的节点数，显然子节点保留可以为0
				dp[u][j]=max(dp[u][j],dp[u][j-k-1]+dp[v][k]+w);
	}
}
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>q;
	for(int i=1;i<n;i++){
		int u,v,w; cin>>u>>v>>w;
		add(u,v,w); add(v,u,w);
	}
	dfs(1,1);
	cout<<dp[1][q]<<endl;
}