洛谷p1273 有线电视网（树上背包经典）

传送门

本题中，对于每个结点，其可连接到的用户数即为后代中叶节点的个数，因此可以先跑一遍 $dfs$ （代码中实现为 $deal（）$ 函数）统计。同时对于任一结点，若将其方案按连接的用户数进行划分，则可选方案为 $1，2，...，cnt[n]$，其中 $cnt[n]$ 即为后代中叶节点的个数。由于一次仅能选择其中一种方案，因此可以考虑采用分组背包进行处理。

采用状态 $f[u][j]$ 表示在当前的结点 $u$ 选择连接 $j$ 个用户能够得到的最大收益（该收益可能为负数，因此需要初始化为一个较小的值，否则产生负贡献时也将取为 $0$ ） ，则当 $u$ 不是叶结点时，可以枚举其每一个儿子 $v$ 连接的用户数 $k$ ，那么剩余的 $j-k$ 个连接的用户将在结点 $u$ 的其它儿子上出现，由此有状态转移方程：$f[u][j]=max(f[u][j],f[v][k]+f[u][j-k]-fare）$，其中 $fare$ 为连接结点 $u$ 和其儿子 $v$ 需要的花费。

由于本题的要求为不亏本，因此最后由大到小枚举根节点连接的用户数时，只要最大收益不小于 $0$ 即为最优答案。

#include <bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;

const int N=3010;
int f[N][N];//当前已经到第i个结点 且这个结点连接了j个用户能够获得的最大收益 最终只要该收益不小于0即可 
vector<PII> h[N];
int w[N],cnt[N];

int n,m;

void dfs(int u){//分组背包 
	if(h[u].size()==0){
		f[u][1]=w[u];
		return ;
	}
	
	for(int i=1;i<=cnt[u];i++)//初始化 防止负贡献取max后为0 
		f[u][i]=-30000;
		
	for(auto x:h[u]){
		int id=x.first;
		dfs(id);
		for(int j=cnt[u];j>=1;j--){
			for(int k=1;k<=min(cnt[id],j);k++){
				f[u][j]=max(f[u][j],f[u][j-k]+f[id][k]-x.second);	
			}
		}
		
	}
}

void deal(int u){
	if(h[u].size()==0){//没有儿子 是叶子结点 
		cnt[u]=1;
		return ;
	}
	for(auto x:h[u]){
		int y=x.first; 
		deal(y);
		cnt[u]+=cnt[y];
	}
}

void solve(){
	int n,m;
	cin>>n>>m;
	int k,a,c;
	
	for(int i=1;i<=n-m;i++){
		cin>>k;
		for(int j=1;j<=k;j++){
			cin>>a>>c;
			h[i].push_back({a,c});
		}
	}
	for(int i=1;i<=m;i++)
		cin>>w[n-m+i];
	deal(1);//预处理每个结点下属的叶子结点的个数 
	dfs(1);

	/*
	测试用 每个结点选择j个叶子时的花费 
	
	for(int i=1;i<=n;i++)
		for(int j=1;j<=cnt[i];j++)
			cout<<"i="<<i<<"j="<<j<<"  "<<f[i][j]<<endl;
	
	*/
	
	for(int i=m;i>=0;i--){
		if(f[1][i]>=0){
			cout<<i;
			break;
		}
	}
}

signed main(){
	int t=1;
	//cin>>t;
	while(t--){
		solve();
	}
	return 0;
}