Apple Tree POJ 2486

题意：Wshxzt从根节点1开始在苹果树上游历，树上的每个节点都会存在apple[i]个苹果，从一个节点到它的邻节点耗费步数1。现在Wshxzt可以步行step步，求她可以得到的最大苹果数量。
题解：dp[0][i][j]表示以i为根节点步行j步，且最终回到i，可以得到的最大苹果数量。dp[1][i][j]表示最终不回到i可以得到的最大苹果数量。

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;

const int maxn =  205;
vector<int>map[maxn];
bool vis[maxn];
int N,K;
int dp[maxn][maxn][2];
int val[maxn];
int Max(int a,int b){
	if(a<b) return b;
	return a;
}

void DFS(int u)
{
	int i,j,k;
	if(vis[u])return ;
	vis[u]=1;
	for(i=0;i<maxn;i++)
		dp[u][i][0]=dp[u][i][1]=val[u];
	for(i=0;i<map[u].size();i++)
	{
		int v=map[u][i];
		if(vis[v])continue;
		DFS(v);
		for(j=K;j>=1;j--)
		{
			for(k=1;k<=j;k++)
			{
				dp[u][j][1]=Max(dp[u][j][1],dp[u][j-k][0]+dp[v][k-1][1]);
				if(k%2==0)
				{
					dp[u][j][1]=Max(dp[u][j][1],dp[u][j-k][1]+dp[v][k-2][0]);
					dp[u][j][0]=Max(dp[u][j][0],dp[u][j-k][0]+dp[v][k-2][0]);
				}
			}
		}
	}
}

int main()
{
	int i,j;
	int a,b;
	while(scanf("%d%d",&N,&K)!=EOF)
	{
		for(i=1;i<=N;i++)
			map[i].clear();
		for(i=1;i<=N;i++)
			scanf("%d",&val[i]);
		for(i=1;i<N;i++)
		{
			scanf("%d%d",&a,&b);
			map[a].push_back(b);
			map[b].push_back(a);
		}
		memset(vis,0,sizeof(vis));
		DFS(1);
		printf("%d\n",Max(dp[1][K][0],dp[1][K][1]));
	}
	return 0;
}