Apple Tree(树形背包)

http://poj.org/problem?id=2486

---

思路：

很多把它看成了分组背包..其实不用

 

用d[i][j]表示从i点开始走j步能够获得的最大价值，

但是这个题不同的地方，就是如果i到另一棵子树合并的时候，是要往回走步数的。

因此这个题要多开一维讨论从i出发走j步之后是否回到i。（因为只有回到i之后，才能再从i向其他地方走。）

因此状态应该设计为d[i][j][0/1]，表示从i出发走j步后是否回到i，0表示回，1表示不回

转移方程：

对于回到i的，和之前的背包一样，不过由于要回到i，所以其要多花两步。一步走到v子树，一步从v子树走回来。

对于不回到i的，有两种情况：一种是先在其他子树走完回到i，再到v里面走不回来。另外一种是先从v走回来，然后在其他乱走。

 

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

注意初始化，对于单个节点，其所有步数下，获得的价值，都是本身。

单更新一个dp[u][0][1]=dp[u][0][0]=val[u]是不行的

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=220;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[maxn][maxn][3];
LL val[maxn];
vector<LL>g[maxn];
LL n,k;
void dfs(LL u,LL fa){
     for(LL j=0;j<=k;j++){
        dp[u][j][0]=dp[u][j][1]=val[u];
     }
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         dfs(v,u);
         for(LL j=k;j>=0;j--){
             for(LL o=0;o<=j;o++){
                 if(o>=2){
                    dp[u][j][0]=max(dp[u][j][0],dp[u][j-o][0]+dp[v][o-2][0]);
                 }
                 if(o>=1){
                    dp[u][j][1]=max(dp[u][j][1],dp[u][j-o][0]+dp[v][o-1][1]);
                    dp[u][j][1]=max(dp[u][j][1],dp[u][j-o][0]+dp[v][o-1][0]);///应该是重复的
                 }
                 if(o>=2){
                    dp[u][j][1]=max(dp[u][j][1],dp[u][j-o][1]+dp[v][o-2][0]);
                 }
             }
         }
     }
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   while(cin>>n>>k){
        memset(val,0,sizeof(val));
        for(LL i=0;i<maxn;i++) g[i].clear();
        memset(dp,0,sizeof(dp));
        for(LL i=1;i<=n;i++) cin>>val[i];
        for(LL i=1;i<n;i++){
            LL u,v;cin>>u>>v;
            g[u].push_back(v);g[v].push_back(u);
        }
        dfs(1,0);
        cout<<max(dp[1][k][0],dp[1][k][1])<<"\n";
   }
   return 0;
}