选课【树形dp】

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
typedef long long ll;
using namespace std;
const int maxn = 2000;
int m, n;
int dp[maxn][maxn], sc[maxn], fa[maxn];
struct edge{
    int x, y, next;
}a[maxn];
int head[maxn], len = 0;
void add(int x, int y){
    a[++len].x = x;
    a[len].y = y;
    a[len].next = head[x];
    head[x] = len;
}
int dfs(int u){
    int cnt = 0; //表示节点 u 下总共有多少门课
    for(int i=head[u]; i; i=a[i].next){ // 枚举 u 的子节点
        int v = a[i].y;
        cnt += dfs(v)+1;              // 更新课程数
        for(int j=min(cnt,m); j>=1; j--){ // 在 u 下选 j 门课， 01背包注意倒序
            for(int k=0; k<j; k++){  // 枚举选择子节点的子节点个数
                dp[u][j] = max(dp[u][j], dp[u][j-k-1]+dp[v][k]+sc[v]);
            }
        }
    }
    return cnt;
}
int main(){
    scanf("%d%d", &m, &n);
    for(int i=1; i<=m; i++){
        scanf("%d%d",&fa[i],&sc[i]);
        add(fa[i],i);
    }
    dfs(0);
    printf("%d\n",dp[0][n]);
    return 0;
}