POJ 2486 Apple Tree

树形dp。。

dp[1][n][k]     表示走了k步在n号结点获得的最大值。

其中第一维   0表示返回本身，1表示不返回。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <map>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <cstdlib>
#include <ctime>
using namespace std;

typedef long long lld;
#define REP(i,n) for(int i=0;i<(n);i++)

const int maxn = 205;
const int inf = 0x7f7f7f7f;
int n,K;
vector<int>vv[maxn];
int app[maxn];
int dp[2][maxn][maxn];//0表示返回本身 1表示不返回
void DP(int u,int p)
{
   for(int i=0;i<=K;i++)
    dp[0][u][i]=dp[1][u][i]=app[u];
   for(int i=0;i<vv[u].size();i++)
   {
       int v=vv[u][i];
       if(v==p) continue;
       DP(v,u);
       for(int k=K;k>=0;k--)
       {
           for(int j=0;j<=k;j++)
           {
               dp[0][u][k+2]=max(dp[0][u][k+2],dp[0][v][j]+dp[0][u][k-j]);
               dp[1][u][k+2]=max(dp[1][u][k+2],dp[0][v][j]+dp[1][u][k-j]);
               dp[1][u][k+1]=max(dp[1][u][k+1],dp[1][v][j]+dp[0][u][k-j]);
           }
       }
   }
}
int main()
{
   // int n,m;
    while(~scanf("%d%d",&n,&K))
    {
        memset(dp,0,sizeof(dp));
        for(int i=0;i<=n;i++) vv[i].clear();
        for(int i=1;i<=n;i++) scanf("%d",&app[i]);
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            vv[u].push_back(v);
            vv[v].push_back(u);
        }
        DP(1,0);
        printf("%d\n",dp[1][1][K]);
    }
    return 0;
}