[洛谷1273] 有线电视网

来源:互联网 发布:datetimeoffset mysql 编辑:程序博客网 时间:2024/04/28 11:43

题目描述

  某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。
  从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
  现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
  写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
这里写图片描述   


输入格式

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


输出格式

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


样例数据

样例输入

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,有线电视网就亏本了,而只让③④两个用户看比赛就不亏本了。


题目分析

树形背包,较难,因为题目没有给出钱的范围,因此不能直接定义状态,需要间接定义状态。
设f[i,j]表示i结点保留j个叶子结点的最大盈利值
故Ans=max i(f[1,i]>=0)
于是统计出叶子结点个数Leaf[i],枚举分给儿子结点和分给后代的叶子结点个数,得到状态转移方程:f[i,j]=max(f[i,j],f[i,i-j]+f[son,j]-dist);
其中j需倒向枚举,因为是01背包非完全背包。
边界:叶子结点f[leaf,1]=money[leaf]


源代码

#include<algorithm>#include<iostream>#include<iomanip>#include<cstring>#include<cstdlib>#include<vector>#include<cstdio>#include<cmath>#include<queue>using namespace std;inline const int Get_Int() {    int num=0,bj=1;    char x=getchar();    while(x<'0'||x>'9') {        if(x=='-')bj=-1;        x=getchar();    }    while(x>='0'&&x<='9') {        num=num*10+x-'0';        x=getchar();    }    return num*bj;}int n,leaf,ans=0,f[3005][3005],Money[3005],Leaf[3005];struct Edge {    int from,to,dist;};vector<Edge>edges;vector<int>G[3005];void AddEdge(int from,int to,int dist) {    edges.push_back((Edge) {        from,to,dist    });    int cnt=edges.size();    G[from].push_back(cnt-1);}void Dfs(int Now) {    if(Now>=n-leaf+1) {        Leaf[Now]=1;        f[Now][1]=Money[Now];        return;    }    for(int i=0; i<G[Now].size(); i++) {        Edge& e=edges[G[Now][i]];        int Next=e.to;        Dfs(Next);        Leaf[Now]+=Leaf[Next];    }    for(int i=1; i<=Leaf[Now]; i++)f[Now][i]=-0x7fffffff/2;}void TreeDp(int Now) {    for(int i=0; i<G[Now].size(); i++) {        Edge& e=edges[G[Now][i]];        int Next=e.to;        TreeDp(Next);        for(int i=Leaf[Now]; i>=1; i--)            for(int j=1; j<=i&&j<=Leaf[Next]; j++)                f[Now][i]=max(f[Now][i],f[Now][i-j]+f[Next][j]-e.dist);    }}int main() {    n=Get_Int();    leaf=Get_Int();    for(int i=1; i<=n-leaf; i++) {        int m=Get_Int();        for(int j=1; j<=m; j++) {            int to=Get_Int(),dist=Get_Int();            AddEdge(i,to,dist);        }    }    for(int i=n-leaf+1; i<=n; i++)Money[i]=Get_Int();    Dfs(1);    TreeDp(1);    for(int i=1; i<=leaf; i++)        if(f[1][i]>=0)ans=i;    printf("%d\n",ans);    return 0;}

0 0