记录编号	602656	评测结果	AAAAAAAAAA
题目名称	4085.外卖	最终得分	100
用户昵称	左清源	是否通过	通过
代码语言	C++	运行时间	0.354 s
提交时间	2025-07-05 13:29:29	内存使用	4.61 MiB

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
const int N=505;
int f[N][N][2],a[N],n,m;
vector<int>G[N];
void add(int x,int y){
	G[x].push_back(y); 
}
// f[x][i][0]=max(f[x][i][0],f[x][i-j][0]+f[y][j-2][0])
// f[x][i][1]=max(f[x][i][1],f[x][i-j][0]+f[y][j-1][1])
// f[x][i][1]=max(f[x][i][1],f[x][i-j][1]+f[y][j-2][0])
void dfs(int x,int fa){
	for(int i=1;i<=m;i++)f[x][i][0]=f[x][i][1]=a[x];
	for(auto y:G[x]){
		if(y==fa)continue;
		dfs(y,x);
		for(int i=m;i>=1;i--){
			for(int j=1;j<=i;j++){
				f[x][i][1]=max(f[x][i][1],f[x][i-j][0]+f[y][j-1][1]);
				if(j>=2){
					f[x][i][0]=max(f[x][i][0],f[x][i-j][0]+f[y][j-2][0]);
					f[x][i][1]=max(f[x][i][1],f[x][i-j][1]+f[y][j-2][0]);
				}
			}
		}
	}
	return;
} 
int main(){
	freopen("food.in","r",stdin);
	freopen("food.out","w",stdout);
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",a+i);
	for(int i=1,u,v;i<n;i++){
		scanf("%d %d",&u,&v);
		add(u,v),add(v,u);
	}
	dfs(1,0);
	printf("%d\n",max(f[1][m][0],f[1][m][1]));
	return 0;
}