题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入输出格式
输入格式:
输入文件的第一行包含两个用空格隔开的整数N和M,其中2≤N≤3000,1≤M≤N-1,N为整个有线电视网的结点总数,M为用户终端的数量。
第一个转播站即树的根结点编号为1,其他的转播站编号为2到N-M,用户终端编号为N-M+1到N。
接下来的N-M行每行表示—个转播站的数据,第i+1行表示第i个转播站的数据,其格式如下:
K A1 C1 A2 C2 … Ak Ck
K表示该转播站下接K个结点(转播站或用户),每个结点对应一对整数A与C,A表示结点编号,C表示从当前转播站传输信号到结点A的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。
输出格式:
输出文件仅一行,包含一个整数,表示上述问题所要求的最大用户数。
solution
树型DP——分组背包
转移方程
for(int j=min(num[cur],m);j;--j) for(int k=min(num[ev],j-1);k>=0;--k) DP[cur][j]=max(DP[cur][j],DP[cur][j-k-1]+DP[ev][k]+e[i].w);//DP[cur][j]表示以cur为节点的子树选则j个节点时的收益
最后只用从大到小枚举j,容易证明DP[1][j]从大到小枚举j是单调不下降的,直到>=0跳出循环输出j就可以了
code
#include<cstdio> #include<algorithm> #include<cstring> using namespace std; const int N=3010; int n,m,EdgeCnt=0; int dp[N][N],val[N],a[N],t[N]; struct Edge{ int to,w,next; }e[N*N]; int read(){ int x=0,f=1;char ch=getchar(); while (ch<'0' || ch>'9'){if (ch=='-')f=-1;ch=getchar();} while ('0'<=ch && ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();} return x*f; } void addedge(int u,int v,int w){ int p=++EdgeCnt; e[p].to=v;e[p].w=w;e[p].next=a[u]; a[u]=p; } int dfs(int u){ if (u>n-m){ dp[u][1]=val[u]; return 1; } int sum=0; for (int p=a[u];p;p=e[p].next){ int v=e[p].to; int tk=dfs(v); for (int j=0;j<=sum;j++)t[j]=dp[u][j]; for (int j=0;j<=sum;j++) for (int k=0;k<=tk;k++) dp[u][j+k]=max(dp[u][j+k],t[j]+dp[v][k]-e[p].w); sum+=tk; } return sum; } int main(){ n=read(),m=read(); memset(dp,~0x3f,sizeof(dp)); for (int u=1;u<=n-m;u++){ int size=read(); for (int j=1;j<=size;j++){ int v=read(),w=read(); addedge(u,v,w); } } for (int i=n-m+1;i<=n;i++) val[i]=read(); for (int i=1;i<=n;i++) dp[i][0]=0; dfs(1); for (int i=m;i>0;i--) if (dp[1][i]>=0){ printf("%d",i); break; } return 0; }