poj2486(从1出发 走k步 最多得到多少苹果 不必回到1点 经典树形背包dp)

题目
题意： 每个点有一些苹果，从1点出发，走k步最多能拿到多少苹果？
思路： 很经典的树形背包dp。

#include<cstdio>
#include<iostream>
#include<cstring>
#define m(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
const int N=200+5;
struct Edge{int to,nex;}edge[N<<1];int head[N],tot;
inline void add(int from,int to){
    edge[++tot]=(Edge){to,head[from]},head[from]=tot;
    edge[++tot]=(Edge){from,head[to]},head[to]=tot;
}
int val[N],dp[N][N][2],n,K;
void dfs(int x,int fa){
    for(int i=0;i<=K;++i) dp[x][i][0]=dp[x][i][1]=val[x];
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].to;
        if(y==fa) continue;
        dfs(y,x);
        for(int j=K;j>=0;--j){
            for(int k=1;k<=j;++k){
                if(k>=2) dp[x][j][0]=max(dp[x][j][0],dp[y][k-2][0]+dp[x][j-k][0]);
                if(k>=1) dp[x][j][1]=max(dp[x][j][1],dp[y][k-1][1]+dp[x][j-k][0]);
                if(k>=2) dp[x][j][1]=max(dp[x][j][1],dp[y][k-2][0]+dp[x][j-k][1]);
            }
        }
    }
}
int main(){
    //freopen("input.txt","r",stdin);
    while(~scanf("%d%d",&n,&K)){
        m(head,0),tot=0;
        for(int i=1;i<=n;++i) scanf("%d",&val[i]);
        for(int i=1,x,y;i<n;++i) scanf("%d%d",&x,&y),add(x,y);
        dfs(1,1);
        printf("%d\n",dp[1][K][1]);
    }
}