poj 2486 Apple Tree(树形背包)

题意：

给出一棵树，树上每个节点都有对应数量的苹果，一个人可以从任何点出发，问如何让这个人在走k步以内得到最多的苹果。

题解：

分析发现，这题对于走树的路径是有往回走和不往回走的情况，并且注意到了可以重复走一条路多次。

那么可以定义这样的状态 dp[root][j][st] 根节点root的树，走了j步状态为st时能得到的最大苹果数。st=0、1。 st=0 表示不回到root节点，st=1的相反。

那么状态方程就出来了：

dp[u][j+2][1]=max(dp[u][j+2][1],dp[u][j-k][1]+dp[v][k][1]);
dp[u][j+1][0]=max(dp[u][j+1][0],dp[u][j-k][1]+dp[v][k][0]);
dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][1]);

这里要注意下初始化，见代码。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
typedef __int64 lld;
#define oo 0x3f3f3f3f
#define maxn 105
#define maxm 205
int dp[maxn][maxm][2];
struct EDGE
{
    int v,next;
}E[maxn<<1];
int head[maxn],tol;
int val[maxn];
int n,K;

void inst()
{
    memset(head,-1,sizeof head);
    tol=0;
    memset(dp,0,sizeof dp);
}

void add_edge(int u,int v)
{
    E[tol].v=v;
    E[tol].next=head[u];
    head[u]=tol++;
}

void tree_dp(int u,int pre)
{
    for(int i=0;i<=K;i++)
        dp[u][i][0]=dp[u][i][1]=val[u];//可以一直从重复走这个点多次
    for(int i=head[u];i!=-1;i=E[i].next)
    {
        int v=E[i].v;
        if(v==pre) continue;
        tree_dp(v,u);
        for(int j=K;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][1]);
                dp[u][j+1][0]=max(dp[u][j+1][0],dp[u][j-k][1]+dp[v][k][0]);//人留在子树中
                dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][1]);//人没留在子树中
            }
    }
}

int main()
{
    int u,v;
    while(scanf("%d%d",&n,&K)!=EOF)
    {
        inst();
        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        for(int i=1;i<=n-1;i++)
        {
            scanf("%d%d",&u,&v);
            add_edge(u,v);
            add_edge(v,u);
        }
        tree_dp(1,-1);
        printf("%d\n",max(dp[1][K][0],dp[1][K][1]));
    }
    return 0;
}