餐馆

餐馆

$既然是棵树，肯定就要dp了$
$dp[s][j],表示在以s为根的子树内走j步的最大价值，但我们发现这样没法统计答案$

$定义dp[s][j][0/1]表示在以s为根的子树内走j步，最终回或不会s的最大价值，然后就可以转移了$

$对于一个v，可以是走完s其它节点再回到s，再走v,也可以是先走v，再走s的价值$

#include<bits/stdc++.h>
using namespace std;

const int N=505;
int dp[N][N][2],n,m,a[N],head[N],cnt=0,size[N];
struct edge{
	int link,v;
}q[N<<1];
void put(int x,int y){
	q[++cnt].v=y;
	q[cnt].link=head[x];
	head[x]=cnt;
}
void dfs(int s,int fa){
	dp[s][1][1]=a[s],size[s]=1;
	for(int i=head[s];i;i=q[i].link){
		int v=q[i].v;
		if(v==fa) continue;
		dfs(v,s);
		size[s]+=size[v];
	}
}
void dfs2(int s,int fa){
    int sum=1;
	for(int i=head[s];i;i=q[i].link){
		int v=q[i].v;
		if(v==fa) continue;
		dfs2(v,s);
		int up=4*sum-4,up2=4*size[v]-4;
		if(sum==1) up=1;
		if(size[v]==1) up2=1;
		up=min(up,m),up2=min(up2,m);
		for(int j=up;j>=0;j--){
			for(int kk=up2;kk>=0;kk--)
			{
			if(kk+j+2<=m)dp[s][kk+j+2][1]=max(dp[s][kk+j+2][1],dp[s][j][1]+dp[v][kk][1]);
			if(kk+j+2<=m)dp[s][kk+j+2][0]=max(dp[s][kk+j+2][0],dp[s][j][0]+dp[v][kk][1]);
			if(kk+j+1<=m)dp[s][kk+j+1][0]=max(dp[s][kk+j+1][0],dp[s][j][1]+max(dp[v][kk][0],dp[v][kk][1]));
		}
	 }
		sum+=size[v];
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		put(u,v),put(v,u);
	}
	dfs(1,0);
	dfs2(1,0);
	printf("%d",max(dp[1][m][0],dp[1][m][1]));
}