| 比赛 | 
    20241128 | 
    评测结果 | 
    AAAAAAAAAA | 
    | 题目名称 | 
    外卖 | 
    最终得分 | 
    100 | 
    | 用户昵称 | 
    darkMoon | 
    运行时间 | 
    0.513 s  | 
    | 代码语言 | 
    C++ | 
    内存使用 | 
    4.87 MiB  | 
    | 提交时间 | 
    2024-11-28 09:57:42 | 
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
using namespace std;
auto IN = freopen("food.in", "r", stdin);
auto OUT = freopen("food.out", "w", stdout);
auto mread = [](){int x;scanf("%lld", &x);return x;};
const int N = 505;
int n = mread(), m = mread(), f[N][N][2], a[N];
vector<int> v[N];
void dfs(int x, int fa){
    for(int y : v[x]){
        if(y == fa){
            continue;
        }
        dfs(y, x);
        for(int i = m; i >= 0; i --){
            // 给 x 授予 i 的时间
            for(int j = 0; j <= m; j ++){
                // 给 y 授予 j 的时间
                if(i - j - 2 >= 0){
                    f[x][i][0] = max(f[x][i][0], f[x][i - j - 2][0] + f[y][j][0]);
                }
                if(i - j - 1 >= 0){
                    f[x][i][1] = max(f[x][i][1], f[x][i - j - 1][0] + f[y][j][1]);
                    f[x][i][1] = max(f[x][i][1], f[x][i - j - 2][1] + f[y][j][0]);
                }
            }
        }
    }
    for(int i = m; i >= 1; i --){
        f[x][i][0] = max(f[x][i][0], f[x][i - 1][0] + a[x]);
        f[x][i][1] = max(f[x][i][1], f[x][i - 1][1] + a[x]);
    }
    // for(int i = 0; i <= m; i ++){
    //     printf("%lld %lld %lld %lld\n", x, i, f[x][i][0], f[x][i][1]);
    // }
    return;
}
signed main(){
    for(int i = 1; i <= n; i ++){
        cin >> a[i];
    }
    for(int i = 1, x, y; i < n; i ++){
        cin >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1, 0);
    printf("%lld\n", f[1][m][1]);
    return 0;
}