[CODEVS1378]选课（树形dp）

最新推荐文章于 2025-01-06 12:32:11 发布

原创最新推荐文章于 2025-01-06 12:32:11 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#CODEVS #dp

题解同时被 2 个专栏收录

1097 篇文章

订阅专栏

245 篇文章

订阅专栏

本文介绍了一种使用动态规划解决子树中选择节点问题的方法。通过递归地从叶子节点向上计算最大值，实现了对以每个节点为根的子树内选择特定数量节点的最大价值的计算。

题目描述

传送门

题解

f[i][j]表示以i为根的子树选j个的最大值。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=305;
const int max_e=max_n*2;
int n,m,x,w[max_n],f[max_n][max_n];
int tot,point[max_n],next[max_e],v[max_e];
inline void add(int x,int y){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;}
inline void dp(int x){
    for (int i=point[x];i;i=next[i]){
      dp(v[i]);
      for (int j=m;j>=1;--j)
        for (int k=1; k<=j; k++)
          f[x][j]=max(f[x][j],f[x][j-k]+f[v[i]][k]);
    }
    if (x) for (int i=m; i>=1; i--) f[x][i]=f[x][i-1]+w[x];
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d%d",&x,&w[i]),add(x,i);
    dp(0);
    printf("%d\n",f[0][m]);
}