树形背包 二叉苹果树（洛谷 P2015）

二叉苹果树

题目描述

有一棵苹果树，如果树枝有分叉，一定是分2叉（就是说没有只有1个儿子的结点）

这棵树共有N个结点（叶子点或者树枝分叉点），编号为1-N,树根编号一定是1。

我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树

现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。

给定需要保留的树枝数量，求出最多能留住多少苹果。

输入格式

第1行2个数，N和Q(1<=Q<= N,1<N<=100)。

N表示树的结点数，Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。

每行3个整数，前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。

每根树枝上的苹果不超过30000个。

输出格式

一个数，最多能留住的苹果的数量。

---

树形背包的基础题；

总的解法没有什么变化，就是要注意一个细节，就是如果要取 i 结点的子树，那么结点 i 与这个子树的连接边 i–j 就必须取；

所以状态转移方程为：dp[u][j]=max{dp[u][j],dp[u][j-k-1]+dp[v][k]+w[i]};

这个减 1 就是关键，这个要和有体积和价值的树上背包联合理解；

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=110;
const int M=2000100;
const int mod=1e9;
int n,m,head[N],cnt;
int dp[N][N];
struct Node{
	int to,nex,w;
}edge[N*2];
void add(int p,int q,int w){
	edge[cnt].w=w;
	edge[cnt].to=q;
	edge[cnt].nex=head[p];
	head[p]=cnt++;
} 
void dfs(int p,int q){
	for(int i=head[p];~i;i=edge[i].nex){
		int v=edge[i].to;
		int w=edge[i].w;
		if(v!=q){
			dfs(v,p);
			for(int j=m;j>=1;j--){
				for(int k=0;k<j;k++){
					dp[p][j]=max(dp[p][j],dp[p][j-k-1]+dp[v][k]+w);
				}
			}
		}
	}
}
int main(){
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		add(a,b,c),add(b,a,c);
	}
	dfs(1,-1);
	cout<<dp[1][m]<<endl;
	return 0;
}