见注释。
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;
}