树形dp——有线电视网

原文链接：https://www.luogu.com.cn/problem/P1273
、

AC代码：

#include<iostream>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
#define inf 1e9 
int dp[3005][3005];
int n,m;
int head[3005],cnt=0,visit[3005];
typedef struct edged{
	int fr,to,next,w;
}edged;
edged edge[2000005];
void add(int u,int v,int w){
	edge[cnt].fr=u;
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt;
	cnt++;
}
vector<int> vec;
int dfs(int now){
	int i,j=0;
	if(head[now]==-1&&now>n-m&&now<=n){
		dp[now][1]=vec[now-(n-m+1)];
		dp[now][0]=0;//cout<<now<<" "<<n-m<<" "<<vec[now-m]<<endl;
		return 1;
	}
	int sum=0;
	for(i=head[now];i!=-1;i=edge[i].next){
		int e=edge[i].to;
		int con=dfs(e);sum+=con;//cout<<sum<<endl;
		for(int p=sum;p>0;p--){
			for(int q=1;q<=con;q++){
				dp[now][p]=max(dp[now][p],dp[now][p-q]+dp[e][q]-edge[i].w);
			}
		}
	}
	return sum;
}
int main(){
	int i,j;
	for(i=0;i<3005;i++){
		for(j=0;j<3005;j++){
			dp[i][j]=-inf;
		}
		dp[i][0]=0;
	}
	memset(visit,0,sizeof(visit));
	memset(head,-1,sizeof(head));
	cin>>n>>m;
	for(i=1;i<=n-m;i++){
		cin>>j;
		for(;j>0;j--){
			int p,q;
			cin>>p>>q;
			add(i,p,q);
		}
	}
	for(i=0;i<m;i++){
		cin>>j;
		vec.push_back(j);
	}
	dfs(1);
	for(i=m;i>=0;i--){
		//cout<<i<<" "<<dp[1][i]<<endl;
		if(dp[1][i]>=0){
			cout<<i<<endl;
			return 0;
		}
	}
	return 0;
}