树形dp。。
dp[1][n][k] 表示走了k步在n号结点获得的最大值。
其中第一维 0表示返回本身,1表示不返回。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <map>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <cstdlib>
#include <ctime>
using namespace std;
typedef long long lld;
#define REP(i,n) for(int i=0;i<(n);i++)
const int maxn = 205;
const int inf = 0x7f7f7f7f;
int n,K;
vector<int>vv[maxn];
int app[maxn];
int dp[2][maxn][maxn];//0表示返回本身 1表示不返回
void DP(int u,int p)
{
for(int i=0;i<=K;i++)
dp[0][u][i]=dp[1][u][i]=app[u];
for(int i=0;i<vv[u].size();i++)
{
int v=vv[u][i];
if(v==p) continue;
DP(v,u);
for(int k=K;k>=0;k--)
{
for(int j=0;j<=k;j++)
{
dp[0][u][k+2]=max(dp[0][u][k+2],dp[0][v][j]+dp[0][u][k-j]);
dp[1][u][k+2]=max(dp[1][u][k+2],dp[0][v][j]+dp[1][u][k-j]);
dp[1][u][k+1]=max(dp[1][u][k+1],dp[1][v][j]+dp[0][u][k-j]);
}
}
}
}
int main()
{
// int n,m;
while(~scanf("%d%d",&n,&K))
{
memset(dp,0,sizeof(dp));
for(int i=0;i<=n;i++) vv[i].clear();
for(int i=1;i<=n;i++) scanf("%d",&app[i]);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
vv[u].push_back(v);
vv[v].push_back(u);
}
DP(1,0);
printf("%d\n",dp[1][1][K]);
}
return 0;
}