洛谷P1273 有线电视网

题目描述

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

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

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

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

输入格式

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

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

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

K \ \ A_1 \ \ C_1 \ \ A_2 \ \ C_2 \ \ \ldots \ \ A_k \ \ C_kK  A1​  C1​  A2​  C2​  …  Ak​  Ck​

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

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

2

说明/提示

样例解释

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

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

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

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

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

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

上代码：

#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;
}