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]);
    }
}