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

题目描述

有一棵二叉苹果树，如果数字有分叉，一定是分两叉，即没有只有一个儿子的节点。这棵树共 个节点，标号 至 ，树根编号一定为 。
我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树，因为树枝太多了，需要剪枝。但是一些树枝上长有苹果，给定需要保留的树枝数量，求最多能留住多少苹果。

输入格式

第1行2个数，N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数，Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数，前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。

输出格式

输出仅一行，表示最多能留住的苹果的数量。

样例

样例输入
5 2
1 3 1
1 4 10
2 3 20
3 5 20
样例输出
21

题目分析

可以很容易的看出，这是一道树形dp的题目。
但仅仅树形dp还是不够的，因为这道题限制了要保留的树枝的数量。

1. 状态表示：f[u][j] //表示以u节点为根的树，保留j个节点后可以保留的最大苹果数
2. 状态计算：状态f[u][j]，可以由f[u][j-k]+f[v][k]+a[u][v]转移过来。
   因此状态转移方程为：f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+a[u][v]);
3. 注意：
   1）f[][]数组一开始要初始化为负无穷。这么初始化是为了保证子节点只能由其父节点转移而来。
   2）最后的结果为f[1][q+1]。因为f[u][j]表示的是保留了j个节点，而题目要求是保留q条边。而 树的边数=节点数+1。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#include <iomanip>
#define LL long long
#define PII pair<int,int>
using namespace std;
const int N=100+5;
vector<int> h[N];
int a[N][N];					//我们可以用邻接矩阵的方式来树a[u][v]=u-v这条树枝上的苹果数
int f[N][N],s[N];				//s[u]表示以u为根节点的树的节点数
void dfs(int u,int fa)
{
	s[u]=1;						
	f[u][1]=0;					//只保留1个节点的情况下，没有边存在，因此f[u][1]=0
	for(int i=0;i<h[u].size();i++)
	{
		int v=h[u][i];			//dfs的基本操作
		if(v==fa) continue;
		dfs(v,u);
		
		s[u]+=s[v];				//s[u]=sum(s[u的所以子节点])
		for(int j=s[u];j>0;j--)	//基础的背包dp
			for(int	k=1;k<=s[v]&&k<=j;k++)
				f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+a[u][v]);
	}
}
int main()
{
	memset(f,-0x3f,sizeof f);
	int n,m;
	scanf("%d %d",&n,&m);
	for(int i=1;i<n;i++)		//建图
	{
		int u,v,w;
		scanf("%d %d %d",&u,&v,&w);
		h[u].push_back(v);
		h[v].push_back(u);
		a[u][v]=a[v][u]=w;
	}
	dfs(1,-1);					//通过递归的方式dp
	printf("%d\n",f[1][m+1]);
	return 0;
}