树形dp复习 树上依赖背包问题

选课

今天又看了一下这道题，竟然AC不了了

自己的学习效率有点低下

要明白本质，搞透彻

#include<bits/stdc++.h>
#define REP(i, a, b) for(register int i = (a); i < (b); i++)
#define _for(i, a, b) for(register int i = (a); i <= (b); i++)
using namespace std;

const int MAXN =  + ;
struct Edge{ int to, next; };
Edge e[MAXN << ];
int head[MAXN], a[MAXN];
int f[MAXN][MAXN], tot, n, m;

void AddEdge(int from, int to)
{
    e[tot] = Edge{to, head[from]};
    head[from] = tot++;
}

void dfs(int u)
{
    for(int i = head[u]; ~i; i = e[i].next)
    {
        int v = e[i].to; dfs(v);
        for(int t = m; t >= ; t--)
            _for(j, , t - ) //注意子树最多选t-1个点
                f[u][t] = max(f[u][t], f[u][t-j] + f[v][j]);
    }
}

int main()
{
    memset(head, -, sizeof(head)); tot = ;
    scanf("%d%d", &n, &m); m++; //这是个技巧，加上选0节点
    _for(i, , n)
    {
        int x; scanf("%d%d", &x, &f[i][]); //注意这里要初始化
        AddEdge(x, i);
    }

    dfs();
    printf("%d\n", f[][m]);

    return ;
}