LibreOJ10154 选课 -------- 树形dp（依赖背包）

题目链接 https://loj.ac/problem/10154

题意：给定n门课程，某些课程有1门先行课，必须先修了先行课才可以选这门课，选择m门课，每门课有学分，求可以获得的最大学分数。

题解：参考 https://www.cnblogs.com/Chen-Jr/p/11205300.html  

依赖背包问题，把每门课看作一个点，依赖关系是连成边，形成一个森林，再建立一个超级原点建立成有根树。

不难发现，这个选取m个课程的过程本质上跟背包的性质差不多，因此我们可以设dp[x][k]为作为以xx为根的子树，现在已经获取了k个课程后的最大的学分。而对于节点x选取了k个课程的状态dp[x][k]，他显然是由他的儿子son[x]选取了jj个课程的状态dp[son[x]][j]转移而来的，固有状态转移方程：dp[x][k]=max(dp[x][k],dp[to][j]+dp[x][k−j])。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 550;
struct Node {
    int to, next;
}q[maxn];
int head[maxn], cnt = 0, val[maxn], dp[maxn][maxn];
int n, m;

void add_edge(int from ,int to) {
    q[cnt].to = to;
    q[cnt].next = head[from];
    head[from] = cnt++;
}

void init() {
    memset(head, -1, sizeof(head));
    cnt = 0;
}

void dfs(int x) {
    dp[x][1] = val[x];
    for(int i = head[x];i != -1;i = q[i].next) {
        int to = q[i].to;
        dfs(to);
        for(int j = m+1;j >= 1;j--) {
            for(int k = 0;k < j;k++) {
                dp[x][j] = max(dp[x][j], dp[to][k]+dp[x][j-k]);
            }
        }
    }
}


int main() {

    scanf("%d%d", &n, &m);
    init();
    for(int i = 1;i <= n;i++) {
        int to;
        scanf("%d%d", &to, &val[i]);
        add_edge(to, i);
    }

    dfs(0);
    printf("%d\n", dp[0][m+1]);

    return 0;
}