poj 2486 树形dp(给定步数 走一棵树，获得的节点值总和的最大值)

#include<cstdio>#include<cstring>#define MAX(x,y) ((x)>(y)?(x):(y))struct node{int to,next;}e[2400];int cnt,n,k,head[220],dp[220][220][2],v[220];void add_edge(int from,int to){e[cnt].to=to;e[cnt].next=head[from];head[from]=cnt++;}void dfs(int u,int v){for(int i=head[u];i!=-1;i=e[i].next){int to=e[i].to;if(to==v)continue;dfs(to,u);for(int j=k;j>0;j--)for(int p=1;p<=j;p++){dp[u][j][0]=MAX(dp[u][j][0],dp[u][j-p][1]+dp[to][p-1][0]);if(p>1){dp[u][j][0]=MAX(dp[u][j][0],dp[u][j-p][0]+dp[to][p-2][1]);dp[u][j][1]=MAX(dp[u][j][1],dp[u][j-p][1]+dp[to][p-2][1]);}}}} int main(){while(~scanf("%d%d",&n,&k)){   for(int i=1;i<=n;i++)   {     scanf("%d",&v[i]);     for(int j=0;j<=k;j++)     dp[i][j][0]=dp[i][j][1]=v[i];       }       cnt=0;   memset(head,-1,sizeof(head));    for(int i=1;i<n;i++)   {      int a,b;      scanf("%d%d",&a,&b);      add_edge(a,b);      add_edge(b,a);       }       dfs(1,-1);       printf("%d\n",dp[1][k][0]);    }}