记录编号 |
602771 |
评测结果 |
AAAAAAAAAA |
题目名称 |
4085.外卖 |
最终得分 |
100 |
用户昵称 |
秋_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;
}