比赛 |
2025暑期集训第4场 |
评测结果 |
AAAAAAAAAA |
题目名称 |
外卖 |
最终得分 |
100 |
用户昵称 |
左清源 |
运行时间 |
0.295 s |
代码语言 |
C++ |
内存使用 |
4.59 MiB |
提交时间 |
2025-07-05 09:17:47 |
显示代码纯文本
#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;
}