比赛 2025暑期集训第4场 评测结果 WWWWWWWWWW
题目名称 外卖 最终得分 0
用户昵称 秋_Water 运行时间 0.415 s
代码语言 C++ 内存使用 4.46 MiB
提交时间 2025-07-05 10:04:41
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N=508;
vector<int>G[N];
int f[N][N][2],a[N];
int n,m;
void dfs(int u,int fa){
	f[u][1][0]=a[u];
	f[u][1][1]=a[u];
	for(auto v:G[u]){
		if(v!=fa){
			dfs(v,u);
			for(int i=1;i<=m;i++){
				for(int j=1;j<=i;j++){
					if(i-j>=2){
						f[u][i][0]=max(f[u][i][0],f[u][i-j-2][0]+f[v][j][0]);
						f[u][i][1]=max(f[u][i][1],f[u][i-j-2][1]+f[v][j][0]);
					}
					f[u][i][1]=max(f[u][i][1],f[u][i-j-1][0]+f[v][j][1]);
				}
			}
		}
	}
}  
int main(){
	freopen("food.in","r",stdin);
	freopen("food.out","w",stdout);		
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<n;i++){
		int a,b;
		cin>>a>>b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	dfs(1,0);
	int ans=0;
	for(int i=1;i<=m;i++){
		ans=max(ans,max(f[1][i][0],f[1][i][1]));
	}
	cout<<ans<<"\n";

	return 0;
}