Codeforces 1249 F. Maximum Weight Subset

传送门

设 $f[x][i]$ 表示 $x$ 的子树中，离 $x$ 最近的选择的节点距离为 $i$ 的合法方案的最大价值

设 $val[x]$ 表示节点 $x$ 的价值，首先有 $f[x][0]=val[x]$

那么考虑子树的合并，有 $f[x][min(i,j+1)]=max(f[x][min(i,j+1)],f[x][i]+f[v][j])$

注意此时 $f[x][i]$ 不能包括 $v$ 的贡献，这个可以搞个 $tmp$ 存一下新的 $f[x]$，最后统一覆盖掉即可

然后答案就是 $f[rt]$ 里面的状态取个最大值

复杂度 $n^3$

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
inline int read()
{
    int x=,f=; char ch=getchar();
    while(ch<''||ch>'') { if(ch=='-') f=-; ch=getchar(); }
    while(ch>=''&&ch<='') { x=(x<<)+(x<<)+(ch^); ch=getchar(); }
    return x*f;
}
const int N=;
int n,m,val[N];
int fir[N],from[N<<],to[N<<],cntt;
inline void add(int a,int b) { from[++cntt]=fir[a]; fir[a]=cntt; to[cntt]=b; }
int f[N][N],tmp[N],ans;
vector <int> son[N];
void dfs(int x,int fa)
{
    f[x][]=val[x];
    for(int i=fir[x];i;i=from[i])
    {
        int &v=to[i]; if(v==fa) continue;
        dfs(v,x); memset(tmp,,sizeof(tmp));
        for(int j=;j<=n;j++)
            for(int k=;k<=n;k++)
                if(j+k+>m) tmp[min(j,k+)]=max(tmp[min(j,k+)],f[x][j]+f[v][k]);
        for(int j=;j<=n;j++) f[x][j]=tmp[j];
    }
}
int main()
{
    n=read(),m=read(); int a,b;
    for(int i=;i<=n;i++) val[i]=read();
    if(n==) { printf("%d\n",val[]); return ; }
    for(int i=;i<n;i++)
        a=read(),b=read(),
        add(a,b),add(b,a);
    dfs(,);
    for(int i=;i<=n;i++) ans=max(ans,f[][i]);
    printf("%d\n",ans);
    return ;
}