记录编号 602771 评测结果 AAAAAAAAAA
题目名称 4085.外卖 最终得分 100
用户昵称 Gravatar秋_Water 是否通过 通过
代码语言 C++ 运行时间 0.681 s
提交时间 2025-07-05 16:47:55 内存使用 4.46 MiB
显示代码纯文本
#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){
	for(int i=0;i<=m;i++){
		f[u][i][0]=f[u][i][1]=-1e9;
	}
	f[u][0][0]=0;
	f[u][0][1]=0;
	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=m;i>=0;i--){
				for(int j=0;j<=i;j++){
					if(i>=j+2){
						f[u][i][1]=max(f[u][i][1],f[u][i-j-2][1]+f[v][j][1]);
						f[u][i][0]=max(f[u][i][0],f[u][i-j-2][0]+f[v][j][1]);
					}					
					if(i>=j+1){
						f[u][i][0]=max(f[u][i][0],f[u][i-j-1][1]+f[v][j][0]);
						f[u][i][0]=max(f[u][i][0],f[u][i-j-1][1]+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=0;i<=m;i++){
		ans=max(ans,max(f[1][i][0],f[1][i][1]));
	}
	cout<<ans<<"\n";

	return 0;
}