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;  
}