比赛 2025暑期集训第4场 评测结果 ATTTTTTTTT
题目名称 外卖 最终得分 10
用户昵称 李奇文 运行时间 26.990 s
代码语言 C++ 内存使用 3.41 MiB
提交时间 2025-07-05 08:48:03
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=505;
int n,m;
int a[N],hd[N],tot,ans,va[N];
struct node{
	int nxt,to;
}e[N*2];
void add(int x,int y){
	++tot;
	e[tot].to=y;
	e[tot].nxt=hd[x];
	hd[x]=tot;
}
void dfs(int u,int t,int sum){
	if(t==m){
		ans=max(ans,sum);
		return;
	}
	for(int i=hd[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(va[u]){
			dfs(v,t+1,sum);
		}else{
			va[u]=1;
			dfs(u,t+1,sum+a[u]);
			va[u]=0;
			dfs(v,t+1,sum);
		}
	}
	return;
}
int main(){
	freopen("food.in","r",stdin);
	freopen("food.out","w",stdout);
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
		add(v,u);
	}
	dfs(1,0,0);
	cout<<ans<<endl;
	return 0;
}