【DP计划】10.14——[洛谷]有线电视网 （树形DP）MEDIUM++

题目描述

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

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

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

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

输入输出格式

输入格式：
输入文件的第一行包含两个用空格隔开的整数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的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。

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

输入输出样例

输入样例#1：
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2
输出样例#1：
2
说明

样例解释

如图所示，共有五个结点。结点①为根结点，即现场直播站，②为一个中转站，③④⑤为用户端，共M个，编号从N-M+1到N，他们为观看比赛分别准备的钱数为3、4、2，从结点①可以传送信号到结点②，费用为2，也可以传送信号到结点⑤，费用为3（第二行数据所示），从结点②可以传输信号到结点③，费用为2。也可传输信号到结点④，费用为3（第三行数据所示），如果要让所有用户（③④⑤）都能看上比赛，则信号传输的总费用为：

2+3+2+3=10，大于用户愿意支付的总费用3+4+2=9，有线电视网就亏本了，而只让③④两个用户看比赛就不亏本了。

---

这是一棵树，所以肯定是树形DP的范畴。但是按照最原始的树形DP思路，也就是从下往上推，并不能解决这道题，因为对于一棵子树，是允许存在小于零的情况的，但又必须保证最上面的不小于0。

于是换一种思路，在树上做背包，对于每个节点做一次背包,dp[i][j]表示节点i，在其子树中选了j个叶子节点时所能获得的最大收益（可能是负数）。那么这其实是一个分组背包问题，它的每个儿子节点就是一个组，而在这个组里只能选择一种情况（这个儿子往下取多少个）。于是就有了转移方程

for(int i=head[x];i!=-1;i=nxt[i]){
        int to=L[i].to;calc(to);
        for(int j=siz[x];j>=0;j--)
          for(int k=1;k<=siz[to];k++)
           if(j>=k) dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[to][k]-L[i].val);
    }

最后统计答案就枚举最大$x$满足的$dp[1][x]>=0$即可。

#include<bits/stdc++.h>
#define MAXN 3005
using namespace std;
int read(){
    char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
int n,m,cnt,ans,head[MAXN],nxt[MAXN],go[MAXN],dp[MAXN][MAXN],w[MAXN],son[MAXN],siz[MAXN];
struct node{
    int to,val;
}L[MAXN];
void add(int x,int y,int c){
    L[cnt]=(node){y,c};
    nxt[cnt]=head[x];head[x]=cnt;cnt++;
}
int dfs(int x){
    siz[x]=1;
    for(int i=head[x];i!=-1;i=nxt[i]){
        int to=L[i].to;
        siz[x]+=dfs(to);
    }
    return siz[x];
}
void calc(int x){
    if(x>n-m){dp[x][1]=w[x];return;}
    for(int i=head[x];i!=-1;i=nxt[i]){
        int to=L[i].to;calc(to);
        for(int j=siz[x];j>=0;j--)
          for(int k=1;k<=siz[to];k++)
           if(j>=k) dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[to][k]-L[i].val);
    }
}
int main()
{
    n=read();m=read();
    memset(head,-1,sizeof(head));
    memset(dp,~0x3f,sizeof(dp));
    for(int i=1;i<=n;i++) dp[i][0]=0;
    for(int i=1;i<=n-m;i++){
        son[i]=read();
        for(int j=1;j<=son[i];j++){
            int x=read(),y=read();add(i,x,y);
        }
    }
    for(int i=1;i<=m;i++) w[n-m+i]=read();
    dfs(1);calc(1);
    for(int i=0;i<=n;i++) if(dp[1][i]>=0) ans=i;
    printf("%d",ans);
    return 0;
}