树形dp-洛谷 P2014 选课

https://www.luogu.org/problem/show?pid=2014
我一开始想不出来，看了题解后却发现是最基本的模型
唉~
这里因为是森林所以我们简单的把森林合并到一个节点0；
f[i][j]表示再i点的子孙里取j个的解；
当然不包括i；

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#define Ll long long
using namespace std;
struct cs{
    int to,next,vv;
}a[3001];
int head[3001],v[3001],f[3001][3001];
int n,m,ll,x,y,z,nn;
void init(int x,int y){
    ll++;
    a[ll].to=y;
    a[ll].next=head[x];
    head[x]=ll;
}
int dfs(int x){
    f[x][0]=v[x];
    if(head[x]==-1)return 1;
    int sum=0,son;
    for(int k=head[x];k!=-1;k=a[k].next){
        son=dfs(a[k].to);
        sum+=son;
        for(int j=sum;j;j--)
        for(int i=0;i<=son;i++)
        if(j-i-1>=0)//这个-1，1代表a[k].to节点本身 
        f[x][j]=max(f[x][j],f[x][j-i-1]+f[a[k].to][i]);
    }
    return sum+1;//这个+1，加上想节点自己 
}
int main()
{
    memset(head,-1,sizeof head);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&x,&v[i]);
        init(x,i);
    }
    nn=dfs(0);
    printf("%d",f[0][m]);
}