洛谷P2014 选课 树形dp

给出$n\le 300$个结点，每个结点都有一个权值，然后一些结点必须只有选了前驱结点这个结点才可以被选。现在最多选$m\le 300$个结点，求最大权值。
实际上这个是一个森林，但是通过设置一个权值为$0$的虚点$0$，把所有的没有前驱的点全部连到这个结点上，并且这个结点是必须要选的，也就是选择$m+1$个点。
$f_{u,i}$表示$u$这个子树选择$i$个结点的最大值。$f_{u,0}=0，f_{u,1}=w_u$。 f u , i = m a x { f u , j + f v , i − j } f_{u,i}=max\{ f_{u,j}+f_{v,i-j} \} fu,i​=max{fu,j​+fv,i−j​}。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=3e2+7;
int f[N][N];
vector<int> go[N];
int w[N],sz[N];  
int n,m;
void dfs(int u,int fa) {
	sz[u]=1;
	for(auto &v:go[u]) {
		if(v==fa) continue;
		dfs(v,u);
		sz[u]+=sz[v];	
		for(int i=min(sz[u],m);i>=1;i--) {
			int up=min(sz[v],i-1);
			for(int j=0;j<=up;j++) {
				f[u][i]=max(f[u][i],f[v][j]+f[u][i-j]);
			} 
		}
	}
}
int main() {
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) {
		int fa,x; 
		scanf("%d%d",&fa,&x);
		go[fa].push_back(i);
		go[i].push_back(fa);
		w[i]=x;
	}
	for(int i=1;i<=n;i++)
		f[i][1]=w[i];
	m++;
	dfs(0,-1);
	printf("%d\n",f[0][m]); 
	return 0;
}