Apple Tree POJ - 2486 树形dp

题意：求用最多用k步采摘最多的苹果数,点u->v算一步,点v->u即回来也算一步 、

参考http://blog.youkuaiyun.com/xingyeyongheng/article/details/17994781
应该是挺经典的了。。我也不知道怎么处理走回来， 的，
没想到是再开一维。。

int val[N];
int dp[N][210][2];
int fst[N],nxt[2*N],vv[N],tot;
int n,k;
void add(int u,int v){
    vv[tot]=v;nxt[tot]=fst[u];fst[u]=tot++;
}
void dfs(int u,int fa){
    for(int i=fst[u];~i;i=nxt[i]){
        int v=vv[i];
        if(v==fa)continue;
        dfs(v,u);
        for(int j=k;j>=0;--j){
            for(int t=1;j+t<=k;++t){
                dp[u][j+t][0]=max(dp[u][j+t][0],dp[u][j][1]+dp[v][t-1][0]+val[v]);
                if(t>=2)dp[u][j+t][0]=max(dp[u][j+t][0],dp[u][j][0]+dp[v][t-2][1]+val[v]);
                if(t>=2)dp[u][j+t][1]=max(dp[u][j+t][1],dp[u][j][1]+dp[v][t-2][1]+val[v]);
            }
        }
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    while(~sf("%d%d",&n,&k)){
        mem(dp,0);mem(fst,-1);tot=0;
        rep(i,1,n)sf("%d",&val[i]);
        rep(i,1,n-1){ int u,v;sf("%d%d",&u,&v);add(u,v);add(v,u); }
        dfs(1,1);
        pf("%d\n",dp[1][k][0]+val[1]);
    }
}