比赛 2025暑期集训第4场 评测结果 WWWWWTTTEE
题目名称 外卖 最终得分 0
用户昵称 rb_tree 运行时间 9.280 s
代码语言 C++ 内存使用 3.54 MiB
提交时间 2025-07-05 10:52:56
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
using namespace std;
int read()
{
    int x=0,w=1;
    char ch=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x*w;
}
void write(int x)
{
    if(x<0)
    {
        x=-x;
        putchar('-');
    }
    if(x>=10) write(x/10);
    putchar((x%10)^48);
}
const int N=105,M=205;
int h[N],nxt[M],to[M],_cnt,n,m,a[N],dp[N][M];
void add_edge(int a,int b)
{
    to[++_cnt]=b;
    nxt[_cnt]=h[a];
    h[a]=_cnt;
}
void dfs(int x,int f)
{
    for(int i=h[x];i;i=nxt[i])
    {
        int vv=to[i];
        if(vv==f) continue;
        dfs(vv,x);
        for(int v=m;v>=1;v--) 
        {
            dp[x][v]=max(dp[x][v],dp[vv][v-1]);
            if(v>=2) dp[x][v]=max(dp[x][v],dp[vv][v-2]+a[vv]);
        }
    }
}
signed main()
{
    freopen("food.in","r",stdin);
    freopen("food.out","w",stdout);
    n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<=m;i++) dp[1][i]=a[i];
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read();
        add_edge(u,v);
        add_edge(v,u);
    }
    dfs(1,0);
    write(max(dp[1][m],dp[1][m-1]+a[1]));
    fclose(stdin);
    fclose(stdout);
    return 0;
}