poj 2486 Apple Tree (带回溯的树形dp）

题意：树上每个节点都有不同个数的苹果，问走k步最多能吃到多少苹果。

分析：

1、和上一题攻打城堡的树形dp比起来，这题同样是有依赖关系的，要走到儿子节点吃苹果就

必须先走到父亲节点。而且都是树上背包问题。但是又是有区别的，这题是可以回溯的。

2、把背包思想应用到这里，把步数看成背包容积v，每个节点的苹果个数看成价值。

dp【0】【i】【j】表示在i为根节点的子树上走j步后回到i点能吃到的最大苹果树。

dp【1】【i】【j】表示在i为根节点的子树上走j步后不回到i点（即停在i的某个子树中的节点上）能吃到的最大苹果树。

那么有三种走法：t为i的一个儿子。

1、从i走到t 再回到i，再走到i的其他子树的结点停下来。

dp[1][i][j+2]=max(dp[1][i][j+2]，dp[0][t][k]+dp[1][i][j-k])  在以t为根的树上走k步，在i的其他子树上走j-k步。

由于i-t,t-i占两步，共j+2步；

2、从i走到除以t为根的其他子树再回溯到i，然后走t子树，停在t子树的某个节点。

dp[1][i][j+1]=max(dp[1][i][j+1]，dp[1][t][k]+dp[0][i][j-k])

3、从i走到t子树然后回到i，再走到其他子树又回到i。

dp[0][i][j+2]=max(dp[0][i][j+2]，dp[0][t][k]+dp[0][i][j-k])

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>

using namespace std;

int dp[2][110][210],val[110],v,num[110];
bool vis[110];
int mp[110][110];

int maxx(int x,int y)
{
    return x>y?x:y;
}

void dfs(int s)
{
    for(int i=0;i<=v;i++)
        dp[0][s][i]=dp[1][s][i]=val[s];
    vis[s]=true;
    for(int i=1;i<=num[s];i++)
    {
        int t=mp[s][i];
        if(!vis[t])
        {
            dfs(t);
            for(int j=v;j>=0;j--)
            {
                for(int k=0;k<=j;k++)
                {
                    dp[1][s][j+1]=maxx(dp[1][s][j+1],dp[0][s][j-k]+dp[1][t][k]);
                    dp[1][s][j+2]=maxx(dp[1][s][j+2],dp[1][s][j-k]+dp[0][t][k]);
                    dp[0][s][j+2]=maxx(dp[0][s][j+2],dp[0][s][j-k]+dp[0][t][k]);
                }
            }
        }
    }
}

int main()
{
    int n,a,b;
    while(scanf("%d%d",&n,&v)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        memset(num,0,sizeof(num));
        memset(dp,0,sizeof(dp));
        memset(mp,0,sizeof(mp));

        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            mp[a][++num[a]]=b;
            mp[b][++num[b]]=a;
        }
        dfs(1);
        printf("%d\n",dp[1][1][v]);
    }
    return 0;
}

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>

using namespace std;

int dp[2][110][210],val[110],v,num[110];
bool vis[110];
//int mp[110][110];
vector<int> mp[110];

int maxx(int x,int y)
{
    return x>y?x:y;
}

void dfs(int s)
{
    for(int i=0;i<=v;i++)
        dp[0][s][i]=dp[1][s][i]=val[s];
    vis[s]=true;
    for(int i=0;i<mp[s].size();i++)
    {
        int t=mp[s][i];
        if(!vis[t])
        {
            dfs(t);
            for(int j=v;j>=0;j--)
            {
                for(int k=0;k<=j;k++)
                {
                    dp[1][s][j+1]=maxx(dp[1][s][j+1],dp[0][s][j-k]+dp[1][t][k]);
                    dp[1][s][j+2]=maxx(dp[1][s][j+2],dp[1][s][j-k]+dp[0][t][k]);
                    dp[0][s][j+2]=maxx(dp[0][s][j+2],dp[0][s][j-k]+dp[0][t][k]);
                }
            }
        }
    }
}

int main()
{
    int n,a,b;
    while(scanf("%d%d",&n,&v)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        for(int i=0;i<=n;i++)
            mp[i].clear();
        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            mp[a].push_back(b);
            mp[b].push_back(a);
        }
        dfs(1);
        printf("%d\n",dp[1][1][v]);
    }
    return 0;
}

试了一下stl的vector和自己写二维数组，stl慢啊= =

记得有一个树形dp题还非得用临界表存图才给过呢。。。。。