Vijos1676 陶陶吃苹果（树形dp）

题目链接

分析：
好好读题啊

如果一个点被保留，那么这个点到根路径上的点都会被保留，前辈说这是：树形依赖问题
解决树形依赖问题的常见做法就是将父节点的值给当前节点，然后用当前节点继续深层的计算

设$f[i][j]$表示在结点$i$到根结点的链一侧选择$j$个结点的最大收益
$f[son]=f[x]$，然后用递归后的$f[son]$更新$f[x]$
$f[x][i]=max(f[son][i−1]+a[son],f[x][i])$

考虑这道题，$t-h<=k$的限制实际上就是白送了你一条链（链上结点数-高度一定等于零）
那么我们可以对于每一条链预处理出链左右两侧的贡献，那么答案实际上就是除了这条链，选择$m$个结点的最大收益

统计答案时，对于一个叶子结点x
枚举左侧选择了$i$个结点（右侧就选择了$m-i$个结点）
该情况的答案即为$f[x][i]+g[x][m-i]+x到根的链总和$

#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

const int N=5010;
int sum[N],n,m,c[N],root,du[N],nxt[N][N];
int f[N][510],g[N][510];

void dfs_1(int now) {
    for (int i=1;i<=du[now];i++) {
        int y=nxt[now][i];
        sum[y]=sum[now]+c[y];             //链 
        for (int j=0;j<=m;j++) f[y][j]=f[now][j];
        dfs_1(y);
        for (int j=1;j<=m;j++) f[now][j]=max(f[now][j],f[y][j-1]+c[y]);
    }
}

void dfs_2(int now) {
    for (int i=du[now];i>=1;i--) {
        int y=nxt[now][i];
        for (int j=0;j<=m;j++) g[y][j]=g[now][j];
        dfs_2(y);
        for (int j=1;j<=m;j++) g[now][j]=max(g[now][j],g[y][j-1]+c[y]);
    }
}

int main() 
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) {
        int x;
        scanf("%d%d",&x,&c[i]);
        if (!x) root=i;
        else du[x]++,nxt[x][du[x]]=i;
    }
    sum[root]=c[root]; 
    dfs_1(root);
    dfs_2(root);
    int ans=0;
    for (int i=1;i<=n;i++) if (!du[i])
        for (int j=0;j<=m;j++)
            ans=max(ans,f[i][j]+g[i][m-j]+sum[i]);
    printf("%d\n",ans);
    return 0;
}