在原来很裸的树形dp上加了些限制条件,不过知道状态转移还是很好做的,但状态转移不好想啊,都是看了人家的结题报告后才知道的!
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<stack>
using namespace std;
#define FRE freopen("a.txt","r",stdin);
#define inf 999999999
#define N 105
vector<int> v[N];
int m,dp[2][N][2*N],vis[N];
void dfs(int now)
{
for(int i=0;i<=m;i++)dp[0][now][i]=dp[1][now][i]=dp[0][now][0]; //在以now为跟的子树里,不管怎样走,其值至少为now的权值
vis[now]=1;
for(int i=0;i<v[now].size();i++)
{
int u=v[now][i];
if(vis[u])continue;
dfs(u);
for(int j=m-1;j>=0;j--)
{
for(int k=m-j;k>=0;k--)
{
dp[0][now][j+k+2]=max(dp[0][now][j+k+2],dp[0][now][j]+dp[0][u][k]); //在以u为跟的子树走了一圈回到now点,然后在除开u的子树再走一圈再回到now
dp[1][now][j+k+1]=max(dp[1][now][j+k+1],dp[0][now][j]+dp[1][u][k]); //在除开u的子树走一圈回到now点,然后在以u为跟的子树再走了一圈但不回到now
dp[1][now][j+k+2]=max(dp[1][now][j+k+2],dp[1][now][j]+dp[0][u][k]); //在以u为跟的子树走了一圈回到now点,然后在除开u的子树在走一圈但不回到now
}
}
}
}
int main()
{
//FRE;
int n;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
scanf("%d",&dp[0][i][0]);
v[i].clear();
}
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
memset(vis,0,sizeof(vis));
dfs(1);
printf("%d\n",dp[1][1][m]);
}
return 0;
}