poj2486 Apple Tree 树形dp好题

由于之前的树形dp只是单向性，不用返回，而这个题有返回的情况。

题意：一颗树，n个点（1-n），n-1条边，每个点上有一个权值，求从1出发，走V步，最多能遍历到的权值

我们把背包的思想用到这里来，做的步数相当于背包的容量，点上的权值相当于价值，给定一定的背包容量，求最多能装进背包的价值

设dp[0][s][j]表示从s（当前根节点）出发，走 j 步，回到s所能获得的最大权值

dp[1][s][j]表示从s（当前根节点）出发，走j步，不回到s所能获得的最大权值

#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <vector>
using namespace std;
vector<int> tree[105];
int a[105];
int dp[2][105][205];
int n,k;
int x,y;
void dfs(int now,int fa)
{
for(int i=0;i<=k;i++)
dp[0][now][i]=dp[1][now][i]=a[now];
for(int p=0;p<tree[now].size();p++)
{
int to=tree[now][p];
if(to==fa) continue;
dfs(to,now);
for(int i=k;i>=0;i--)
{
for(int j=0;j<=i;j++)
{
dp[1][now][i+1]=max(dp[1][now][i+1],dp[0][now][j]+dp[1][to][i-j]);
dp[1][now][i+2]=max(dp[1][now][i+2],dp[1][now][j]+dp[0][to][i-j]);
dp[0][now][i+2]=max(dp[0][now][i+2],dp[0][now][j]+dp[0][to][i-j]);
}
}
}
}
int main()
{
while(~scanf("%d%d",&n,&k))
{
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++) scanf("%d",&a[i]),tree[i].clear();
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
tree[x].push_back(y);
tree[y].push_back(x);
}
dfs(1,0);
printf("%d\n",dp[1][1][k]);
}
return 0;
}