poj 2486 树形DP

见注释。

AC代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

int N, K;
int dp[2][101][201];
vector<int> edge[200];
bool visit[101];
int apple[101];

inline int max( int a, int b ){
	return ( a > b ? a : b );
}

void DFS( int n ){
	if( visit[n] ){
		return;
	}
	visit[n] = true;
	for( int i = 0; i <= K; i++ ){//当不往下走时
		dp[0][n][i] = dp[1][n][i] = apple[n];
	}
	for( int i = 0; i < edge[n].size(); i++ ){//继续往下走 DP
		int t = edge[n][i];
		if( visit[t] ){
			continue;
		}
		DFS( t );
		for( int j = K; j >= 0; j-- ){
			for( int k = 0; k <= j; k++ ){
				dp[0][n][j+2] = max( dp[0][n][j+2], dp[0][n][j-k] + dp[0][t][k] );//在t中走k步获得的最大再回到n
				dp[1][n][j+1] = max( dp[1][n][j+1], dp[0][n][j-k] + dp[1][t][k] );//在t中走k步或得最大，留在t的子树或t
				dp[1][n][j+2] = max( dp[1][n][j+2], dp[0][t][k] + dp[1][n][j-k] );//在t中走k步或得最大，再回到n，再留在n的子树或n
			}
		}
	}
}

int main(){
	while( scanf( "%d%d", &N, &K ) != EOF ){
		
		memset( dp, 0, sizeof( dp ) );
		memset( visit, false, sizeof( visit ) );
		for( int i = 0; i < 200; i++ ){
			edge[i].clear();
		}
		
		for( int i = 1; i <= N; i++ ){
			cin >> apple[i];
		}
		for( int i = 1; i < N; i++ ){
			int temp1, temp2;
			cin >> temp1 >> temp2;
			edge[temp1].push_back( temp2 );
			edge[temp2].push_back( temp1 );
		}
		DFS(1);
		cout << dp[1][1][K] << endl;
	}
	return 0;
}