因为苹果可能在不同的子树中，所以，很容易想到设状态dp_back[i][j]为以i点为树根走j步并回到i点的最大苹果数与dp_to[i][j]不回到i点的两个状态。

于是，转移方程就很明显了。只是注意要减去一来一回，或者不回的边。树形DP里套背包。

但这题远比这复杂，个人认为。因为在实现上细节太多。

 

实现代码1：

1 #include 
      
        2 #include 
       
         3 #include 
        
          4 using namespace std; 5  6 const int MAX=105; 7 vector
         
          G[MAX]; 8 int num[MAX],dp_back[MAX][MAX*2],dp_to[MAX][MAX*2]; 9 int tback[MAX*2],tto[MAX*2];10 int n,s;11 12 void init(){13     for(int i=1;i<=n;i++)14     G[i].clear();15     memset(dp_back,0,sizeof(dp_back));16     memset(dp_to,0,sizeof(dp_to));17 }18 19 void dfs(int u,int f){20     int size=G[u].size();21     for(int i=0;i
          
           =0){30                         tback[p]=max(tback[p],dp_back[u][p-k-2]+dp_back[v][k]+num[v]);31                     }32                     if(p-k-1>=0){33                         tto[p]=max(tto[p],dp_back[u][p-k-1]+dp_to[v][k]+num[v]);34                     }35                     if(p-k-2>=0){36                         tto[p]=max(tto[p],dp_to[u][p-k-2]+dp_back[v][k]+num[v]);37                     }38                 }39             }40             for(int j=0;j<=s;j++){41                 dp_back[u][j]=tback[j];42                 dp_to[u][j]=tto[j];43             }44         }45     }46 }47 48 int main(){49     int u,v;50     while(scanf("%d%d",&n,&s)!=EOF){51         init();52         for(int i=1;i<=n;i++)53         scanf("%d",&num[i]);54         for(int i=1;i
          
         
        
       
      

这是我WA了N久后看了别人的改过来的，每次在DP时才把根节点的值加上。说不清为什么，但是对了。

 

另一个是我原本WA的代码，可以把OJ的讨论板所有数据都过了，但依然WA，后来研究了好久，发现自己代码上的一个问题，那就是当最大步数超过边数的两倍时，就会出现问

题，于是，我只好投机一点，最后扫描一次结果值来获得正确值了。

1 #include 
      
        2 #include 
       
         3 #include 
        
          4 using namespace std; 5  6 const int MAX=105; 7 vector
         
          G[MAX]; 8 int num[MAX],dp_back[MAX][MAX*2],dp_to[MAX][MAX*2]; 9 int tback[MAX*2],tto[MAX*2];10 int n,s;11 12 void init(){13     for(int i=1;i<=n;i++)14     G[i].clear();15     memset(dp_back,0,sizeof(dp_back));16     memset(dp_to,0,sizeof(dp_to));17 }18 19 void dfs(int u,int f){20     int size=G[u].size();21     dp_back[u][0]=num[u];22     dp_to[u][0]=num[u];23     for(int i=0;i
          
           =0){32                         tback[p]=max(tback[p],dp_back[u][p-k-2]+dp_back[v][k]);33                     }34                     if(p-k-1>=0){35                         tto[p]=max(tto[p],dp_back[u][p-k-1]+dp_to[v][k]);36                     }37                     if(p-k-2>=0){38                         tto[p]=max(tto[p],dp_to[u][p-k-2]+dp_back[v][k]);39                     }40                 }41             }42             for(int j=0;j<=s;j++){43                 dp_back[u][j]=tback[j];44                 dp_to[u][j]=tto[j];45             }46         }47     }48 }49 50 int main(){51     int u,v;52     while(scanf("%d%d",&n,&s)!=EOF){53         init();54         for(int i=1;i<=n;i++)55         scanf("%d",&num[i]);56         for(int i=1;i
          
         
        
       
      

 

两个代码除了初始化的位置不一样，其他都是一样的。但我感觉代码2更符合本来的转移方程，你看一下初始化的位置就明白。但最终问题时，不能处理好，那就是当最大步数超过边数的两倍时问题，因为我在初始化时就认为这是一种不可能的情况了。。。

所以，请路过大牛给指点，以去掉最后的扫描一次得到结果。。。