Luogu P1273 有限电视网【树形Dp/树形背包】

题目描述

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

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

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

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

输入输出格式

输入格式：

输入文件的第一行包含两个用空格隔开的整数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，有线电视网就亏本了，而只让③④两个用户看比赛就不亏本了。

状态设计：设f[i][j]是在以i为根的子树上，让j个用户满足需求的最大收益（利润-成本=毛利小五郎）

目标状态：我们从m倒序枚举满足收益大于0（不亏本）时就停止枚举，此时的答案最优。

转移：每次从当前根节点出发，遍历一遍自己的各个子树，每次遍历统计这个子树上有多少人。做一遍树上背包，先倒序再正序。

（好像很多树上背包都是这样的吧）

        for(int j=sum;j>=0;j--)
         for(int k=0;k<=j;k++)//取这个儿子上的几个点 
          f[x][j]=max(f[x][j],f[y][k]+f[x][j-k]-edge[i].val);

code

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int inf=1e9+7;
int n,m,k,tot,sum,y,z;
int head[7000],f[7000][7000];
struct node{
    int to,val,next;
}edge[7000];
void init()
{
    for(int i=1;i<=n;i++)
     for(int j=1;j<=m;j++)
      f[i][j]=-inf;
}
void add(int x,int y,int z)
{
    edge[++tot].val=z;
    edge[tot].to=y;
    edge[tot].next=head[x];
    head[x]=tot;
}
int TDP(int x)
{
    f[x][0]=0;
    if(x>n-m) return 1;//搜到了用户终端 ,+1
    int sum=0;
    for(int i=head[x];i;i=edge[i].next)
    {
        int y=edge[i].to;
        sum+=TDP(y);//统计子树上有多少人 
        for(int j=sum;j>=0;j--)
         for(int k=0;k<=j;k++)//取这个儿子上的几个点 
          f[x][j]=max(f[x][j],f[y][k]+f[x][j-k]-edge[i].val);
    }
    return sum;
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=n-m;i++)
    {
        scanf("%d",&k);
        for(int j=1;j<=k;j++)
        {
            scanf("%d%d",&y,&z);
            add(i,y,z);
        }
    }
    for(int i=n-m+1;i<=n;i++) scanf("%d",&f[i][1]);
    TDP(1);
    for(int i=m;i>=0;i--)
    {
        if(f[1][i]>=0)//f[i][j]以i为根，j个人能看到 的收益 
        {//如果不亏本 
            printf("%d",i);
            return 0;
        }
    }
    return 0;
}

小结：树上跑背包与普通的树形dp相似，遍历完子树后，跑背包，两层循环，外层枚举当前根，逆序；内层枚举子树，正序。