poj 2486

题目大意：

一个人喜欢吃苹果，一个树上的结点都放这苹果，但是这个人最多只能走m步，问走m步可以吃的苹果最多是多少？

解题思路：

这道题目和普通的树形背包有点区别，访问每个结点是按步数算的，这样，访问了子孙结点还可能返回来，也有可能不回来，我们可以多增加一维来处理这样的情况；

dp[i][j][1]表示从i出发最后没有停留在i；

dp[i][j][0]表示从i出发最后停留在i；

对于第一种情况，还要分俩种情况，对于结点i，已及当前要转移状态的v结点，我们可以先访问除了v以外的与i连接的子孙结点，最后回到i，然后再访问v不反回i；另外一种情况则为我们先访问v及其子孙，然后回到i，在访问除了v以外的其他子孙，不回到i；

即状态转移方程为dp[u][j+2][1]=max(dp[u][j+2][1],dp[u][j-k][1]+dp[v][k][0]); //加2是因为：对于v结点，我们需要k步走完回到v，dp【i】【j】不考虑i与父亲结点的关系，我们要先访问v，因此需要额外的俩步从i走到v，再走回i，对于i的其他子树，我们j-k步走完不回来
dp[u][j+1][1]=max(dp[u][j+1][1],dp[u][j-k][0]+dp[v][k][1]);这个是先访问除了v以外其他子树，然后回到i，然后我们需要花费1步走到v，然后很容易证明，从v走k步不回到v获得的价值最大；

第二种情况，我们无论怎么走都得回到i；

及状态转移方程为；

dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][0]);//这个无论先从哪个子树走，都得回来，而且需要计算从i到v走的来回俩步；所以转移方程就一个；

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 210
int cnt,head[maxn];
int n,m;
int dp[maxn][maxn][2],val[maxn];
struct Edge{
	int to,next;
}edge[2*maxn];
void init()
{
	cnt=0;
	memset(dp,0,sizeof(dp));
	memset(val,0,sizeof(val));
	memset(head,-1,sizeof(head));
}
void addedge(int u,int v)
{
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void tree_dp(int u,int f)
{
	for(int i=0;i<=m;i++)
	dp[u][i][1]=dp[u][i][0]=val[u];
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=f)
		{
			tree_dp(v,u);
			for(int j=m;j>=0;j--)
			for(int k=0;k<=j;k++)
			{
				dp[u][j+2][1]=max(dp[u][j+2][1],dp[u][j-k][1]+dp[v][k][0]);
				dp[u][j+1][1]=max(dp[u][j+1][1],dp[u][j-k][0]+dp[v][k][1]);
				dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][0]);
			}
		}
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		init();
		for(int i=1;i<=n;i++)
		scanf("%d",&val[i]);
		for(int i=1;i<n;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			addedge(a,b);
			addedge(b,a);
		}
		tree_dp(1,0);
		printf("%d\n",dp[1][m][1]);
	}
}