poj 1155 树形dp

#include<cstdio>
#include<cstring>
#define INF 10000000
#define MAX(x,y) ((x)>(y)?(x):(y))
struct node
{
	int to,next,v;
}e[3200];
int cnt,head[3200],f[3200],n,m,tot,vis[3200]; 
int dp[3200][3200];
void add_edge(int from,int to,int v)
{
      e[cnt].to=to;
      e[cnt].v=v;
      e[cnt].next=head[from];
      head[from]=cnt++;
}
void dfs(int u,int v)
{
//	printf("begin u=%d v=%d\n",u,v);
	if(vis[u])
	f[u]=1;
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int to=e[i].to,v=e[i].v;
		dfs(to,u);
		f[u]+=f[to];
		for(int j=f[u];j>=0;j--)
		{
			for(int k=0;k<=j;k++)
			{
				dp[u][j]=MAX(dp[u][j],dp[u][j-k]+dp[to][k]-v);
			}
		}
//		for(int j=0;j<=m;j++)
//		{
//			printf("%d ",dp[u][j]);
//		} 
//		printf("\n"); 
//		printf("end  f=%d\n",f[u]);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	tot=n-m;
	memset(head,-1,sizeof(head));
	cnt=0; 
	for(int i=1;i<=n-m;i++)
	{
		int t,a,b;
		scanf("%d",&t);
		while(t--)
		{
			scanf("%d%d",&a,&b);
			add_edge(i,a,b);
		}
	}
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
	{
		dp[i][0]=0;
		for(int j=1;j<=m;j++)
		dp[i][j]=-INF;
	}
	for(int i=tot+1;i<=n;i++)
	{
		scanf("%d",&vis[i]);
		dp[i][1]=vis[i];
	}
	dfs(1,-1);
	for(int i=m;i>=0;i--)
	if(dp[1][i]>=0)
     { 
		printf("%d\n",i);
		break;
	}
}