poj 2486 Apple Tree (树形dp)

定义dp[u][j][1]表示在以u为根的子树中走了j步并返回了u点所能获得的最大的价值，dp[u][j][0]表示在以u为根的子树中走了j步但没有返回u点所能获得的最大价值。

#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#define N 105
using namespace std;

int n,k,weight[N],dp[N][205][2];
vector<int>edge[N];

void dfs(int u,int fa)
{
        int i,len=edge[u].size(),v,j,l;
        dp[u][0][1]=weight[u];
        for(i=0;i<len;i++){
                v=edge[u][i];
                if(v==fa) continue;
                dfs(v,u);
                for(j=k;j>0;j--){
                        for(l=0;j-l-2>=0;l++){
                                dp[u][j][1]=max(dp[u][j][1],dp[u][j-l-2][1]+dp[v][l][1]);
                                dp[u][j][0]=max( dp[u][j][0]  ,  max( dp[u][j-l-1][1]+dp[v][l][0] , max( dp[u][j-l-2][0] + dp[v][l][1] , dp[u][j-l-1][1] + dp[v][l][1]) ) ) ;
                        }
                        dp[u][j][0]=max(dp[u][j][0] , max(dp[u][0][1]+dp[v][j-1][0] , dp[u][0][1] + dp[v][j-1][1]) );
                }
        }
}

int main()
{
        while(~scanf("%d%d",&n,&k)){
                int i;
                memset(dp,-50,sizeof(dp));
                for(i=1;i<=n;i++) scanf("%d",weight+i);
                for(i=1;i<=n;i++) edge[i].clear();
                for(i=1;i<n;i++){
                        int a,b;
                        scanf("%d%d",&a,&b);
                        edge[a].push_back(b);
                        edge[b].push_back(a);
                }
                dfs(1,0);
                int ans=0;
                for(i=k;i>=0;i--){
                        ans=max(ans,max(dp[1][i][0],dp[1][i][1]));
                }
                cout<<ans<<endl;
        }
        return 0;
}