[Luogu] P2015 二叉苹果树

算是树上分组背包的模板题了吧。
先说说分组背包：有n个组，要从每个组中选一个物品。求在容量为v的状态下能获得的最大值。

//dp[i][j] 表示前i组,体积为j的状态下的最大价值
//n[i]表示第i组的大小
//v[i][k]表示第i组内第k个物品的体积
//w[i][k]表示第i组内第k个物品的价值
for (int i=1; i<=N; i++)  //外层枚举组
	for (int k=1; k<=n[i]; k++)  //中层枚举一组里面的物品
		for (int j=v; j>=0; j--)  //内层枚举体积
		dp[i][j]=max(dp[i][j],dp[i-1][j-v[i][k]]+w[i][k])

dp[i][j]只与dp[i-1][]有关，所以可以压缩到一维：

for (int i=1; i<=N; i++)
	for (int k=1; k<=n[i]; k++)
		for (int j=v; j>=0; j--)
		dp[j]=max(dp[j],dp[j-v[i][k]]+w[i][k])

那么对于本题：
1.建一棵有根树
2.将边权值移到儿子结点上
3.树上dp，转移方程dp[x][j]=max{dp[x][k]+dp[son][j-k]}
dp[x][j]表示在以x为根的树上选j个点所能获得的最大值，那么结点x的每一棵子树可以看成一个个组，每一组中取的结点数就是物品。以x为根的树上选j个点就是由个个子树上选的结点数构成的。
tips：由于本题是在苹果树上剪枝，所以在以x为跟的树上选j个结点,j>1时根节点必须选择，即dp[x][j]不能从dp[x][0]+dp[son][j]转移过来。还有就是枚举j的时候要倒着枚举。
具体体现为下面的代码片段

for (int j=size[x]; j>=2; j--)  //倒序枚举
{
	for (int k=1; k<=j; k++)  
		dp[x][j]=max(dp[x][j],dp[x][k]+dp[son][j-k]);
}

完整代码

#include<bits/stdc++.h>
#define FAST ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define INF 0x3f3f3f3f
typedef long long ll;
const int maxn = 1e5+5;

using namespace std;

int n,q;
int dp[1001][1001],w[maxn],size[maxn];
int cnt,head[maxn];
struct e
{
	int to;
	int next;
}edge[maxn];

void add(int from, int to)
{
	edge[++cnt].to=to;
	edge[cnt].next=head[from];
	head[from]=cnt;
}

void dfs(int x)
{
	size[x]=1;
	for (int i=head[x]; i; i=edge[i].next)
	{
		int son=edge[i].to;
		dfs(son); 
		size[x]+=size[son];
		for (int j=size[x]; j>=2; j--)
		{
			for (int k=1; k<=j; k++)
			dp[x][j]=max(dp[x][j],dp[x][k]+dp[son][j-k]);
		}
	}	
}

int main()
{
   	FAST;
	cin>>n>>q;
	q++;
	for (int i=1; i<=n-1; i++)
	{
		int x,y,z;
		cin>>x>>y>>z;
		if (!w[y]) { add(x,y);w[y]=z; }
		else { add(y,x);w[x]=z; }
	}
	for (int i=1; i<=n; i++) dp[i][1]=w[i];
	dfs(1);
	cout<<dp[1][q]<<endl;
return 0;
}