携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第21天，点击查看活动详情

有线电视网

题目描述

某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，树叶为各个用户终端，其他中转站为该树的内部节点。

从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，一场转播的总费用等于传输信号的费用总和。

现在每个用户都准备了一笔费用想观看这场精彩的足球比赛，有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。

写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。

输入格式

输入文件的第一行包含两个用空格隔开的整数 $N$ 和 $M$ ，其中 $2 \le N \le 3000$ ， $1 \le M \le N-1$ ， $N$ 为整个有线电视网的结点总数， $M$ 为用户终端的数量。

第一个转播站即树的根结点编号为 $1$ ，其他的转播站编号为 $2$ 到 $N-M$ ，用户终端编号为 $N-M+1$ 到 $N$ 。

接下来的 $N-M$ 行每行表示—个转播站的数据，第 $i+1$ 行表示第 $i$ 个转播站的数据，其格式如下：

$K \ \ A_1 \ \ C_1 \ \ A_2 \ \ C_2 \ \ \ldots \ \ A_k \ \ C_k$

$K$ 表示该转播站下接 $K$ 个结点（转播站或用户），每个结点对应一对整数 $A$ 与 $C$ ， $A$ 表示结点编号， $C$ 表示从当前转播站传输信号到结点 $A$ 的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。单次传输成本和用户愿意交的费用均不超过 10。

输出格式

输出文件仅一行，包含一个整数，表示上述问题所要求的最大用户数。

样例 #1

样例输入 #1

样例输出 #1

提示

样例解释

如图所示，共有五个结点。结点 ① 为根结点，即现场直播站，② 为一个中转站，③④⑤ 为用户端，共 $M$ 个，编号从 $N-M+1$ 到 $N$ ，他们为观看比赛分别准备的钱数为 $3$ 、 $4$ 、 $2$ 。

从结点 ① 可以传送信号到结点 ②，费用为 $2$ ；

也可以传送信号到结点 ⑤，费用为 $3$ （第二行数据所示）；

从结点 ② 可以传输信号到结点 ③，费用为 $2$ ；

也可传输信号到结点 ④，费用为 $3$ （第三行数据所示）。

如果要让所有用户（③④⑤）都能看上比赛，则信号传输的总费用为： $2+3+2+3=10$ ，大于用户愿意支付的总费用 $3+4+2=9$ ，有线电视网就亏本了，而只让 ③④ 两个用户看比赛就不亏本了。

#include<bits/stdc++.h>
using namespace std;
const int N=3001;
struct node{
	int to,w,nxt;
	node(int a=0,int b=0,int c=0):to(a),w(b),nxt(c){}
} g[N];
int fir[N],cnt,money[N],n,m,dp[N][N];
void add(int f,int t,int w){
	g[++cnt]=node(t,w,fir[f]);
	fir[f]=cnt;
}
int dfs(int u){
	if(u>n-m){//u是一个叶子节点 
		dp[u][1]=money[u];//叶子节点则选一个节点得到最大利益一定是该节点出的钱 
		dp[u][0]=0;
		return 1;
	}
	dp[u][0]=0;//不选任何节点一定收益为0 
	int leafs=0;//统计以u为根的子树中叶子的数目 
	for(int i=fir[u];i;i=g[i].nxt){
		int v=g[i].to,w=g[i].w;
		int t=dfs(v);//t是v为根的子树中的叶子节点数
		leafs+=t;
		for(int j=leafs;j>0;j--)//在前i个子树中选，因此这里只需要知道前i个子树的总结点数，而不是u的所有节点个数 
			for(int k=0,kk=min(j,t);k<=kk;k++)//v最多分到的节点不可能超过他的叶子数，也不能超过其所在子树节点总数 
				dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-w);
	}
	return leafs;
}
int main(){
	int x,t,w;
	cin>>n>>m;
	for(int i=1;i<=n-m;i++){
		cin>>x;
		for(int j=0;j<x;j++){
			cin>>t>>w;
			add(i,t,w);
		}
	}
	for(int j=n-m+1;j<=n;j++)
		cin>>money[j];
	memset(dp,0x8f,sizeof(dp));
	memset(dp,0,sizeof(dp[0]));
	dfs(1);
	for(int i=m;i>=0;i--){//将用户反向枚举，一旦发现非负数就输出
		if(dp[1][i]>=0){
			cout<<i<<endl;
			break;
		}
	} 
	return 0;
}

【动态规划】有线电视网