Luogu P1273 有线电视网(树形dp+背包)

P1273 有线电视网

题面

题目描述

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

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

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

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

输入输出格式

输入格式：

输入文件的第一行包含两个用空格隔开的整数 \(N\) 和 \(M\) ，其中 \(2 \leq N \leq 3000\) ， \(1 \leq M \leq 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 … A_k \ C_k
\]

\(K\) 表示该转播站下接 \(K\) 个结点(转播站或用户)，每个结点对应一对整数 \(A\) 与 \(C\) ， \(A\) 表示结点编号， \(C\) 表示从当前转播站传输信号到结点 \(A\) 的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。

输出格式：

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

输入输出样例

输入样例：

5 3
2 2 2 5 3
2 3 2 4 3
3 4 2

输出样例：

2

说明

样例解释

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

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

思路

肝试炼场咯! -Uranus

设 \(dp[i][j]\) 表示在 \(i\) 节点满足 \(j\) 位观众的最大收益，那么我们就可以用 \(i\) 节点的每一个儿子节点来更新。

\[dp[i][j]=max \{ dp[i][j-k]+dp[w][k]-cost[i][w](w \in son[i]) \}
\]

然后就在 \(dfs\) 的过程中做这样的动态规划就好啦。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int MAXN=3005;
const int INF=1000000;
int n,m,ans,cnt[MAXN],sz[MAXN],son[MAXN][MAXN],c[MAXN][MAXN],dp[MAXN][MAXN];
int read()
{
    int re=0;
    char ch=getchar();
    while(!isdigit(ch)) ch=getchar();
    while(isdigit(ch)) re=(re<<3)+(re<<1)+ch-'0',ch=getchar();
    return re;
}
void dfs(int now)
{
    for(int i=1;i<=cnt[now];i++)
    {
        dfs(son[now][i]);
        sz[now]+=sz[son[now][i]];
        for(int k=sz[now];k>=0;k--)
            for(int j=0;j<=sz[son[now][i]];j++)
                dp[now][k]=max(dp[now][k],dp[now][k-j]+dp[son[now][i]][j]-c[now][i]);
    }
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            dp[i][j]=-INF;
    for(int i=1;i<=n-m;i++)
    {
        cnt[i]=read();
        for(int j=1;j<=cnt[i];j++) son[i][j]=read(),c[i][j]=read();
    }
    for(int i=n-m+1;i<=n;i++) sz[i]=1,dp[i][1]=read();
    dfs(1);
    for(int i=m;i>=0;i--)
        if(dp[1][i]>=0)
        {
            printf("%d",i);
            return 0;
        }
}