洛谷 1273 有线电视网

原题：https://www.luogu.org/problem/show?pid=1273

题目描述

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

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

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

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

输入格式

输入文件的第一行包含两个用空格隔开的整数N和M，其中$2≤N≤3000，1≤M<N$，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

题解

本题是树形依赖背包问题。

设 $f[u,i]$ 表示以 u 为根节点的子树中，选择 i 个客户后最大的利润，则：

$$f[u,i] = \max\limits_{0 \leq i \leq j}_{v \in child(u)}(f[u,i],f[u,i-j]+f[v,j]-W(u,v))$$

其中的$W(u,v)$为从u到v的边权，也是成本之一。

具体可以看下图：

目标：$\text{ans} = f[1,i] \geq 0，其中i为最大值。$

边界：$对于所有节点 u，均有 f[u,0]=0。若 u 为叶节点，有 f[u,1]=money(v)。其余均为-\infty$。
这里的money()为用户提供的钱数，在程序中用w[]表示。

转移的话，先搜一下子树有几个叶子，然后枚举儿子。
枚举当前儿子分几个叶子，这里的枚举顺序值得注意。
从大到小枚举i分几个，从小到大枚举j分几个。
避免重复选择。

代码

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cctype>
using namespace std;

const int NN = 3001;

struct Edge {
    int to, pw, ne;
    Edge (int to = 0, int pw = 0, int ne = 0) :
        to(to), pw(pw), ne(ne) {}
} edge[NN];
//有向边就可以解决问题。

int head[NN], cnt;
int n, m;
int f[NN][NN];
int w[NN];
int s[NN];
//s[x]用来统计以x为根的子树有几个用户。

inline int read() {
    int r = 0;
    char cc = getchar();
    while (!isdigit(cc)) cc = getchar();
    while (isdigit(cc)) {
        r = r * 10 + cc - 48;
        cc = getchar();
    }
    return r;
}

inline void addEdge(int x, int y, int z) {
    edge[++cnt] = Edge(y, z, head[x]);
    head[x] = cnt;
}

void dfs(int p) {
    f[p][0] = 0;
    for (int e = head[p]; e; e = edge[e].ne) {
        int v = edge[e].to;
        dfs(v);
        s[p] += s[v];
        for (int i = s[p]; ~i; i--) {//注意是倒循环
            for (int j = 0; j <= i; j++) {//儿子最多选它后代中的用户个数
                f[p][i] = max(f[p][i], f[p][i - j] + f[v][j] - edge[e].pw);
            }
        }
    }
}

int main() {
    n = read();
    m = read();
    memset(f, 0xc0, sizeof f);
    for (int i = 1; i < n - m + 1; i++) {
        int k = read();
        while (k--) {
            int A = read();
            int C = read();
            addEdge(i, A, C);
        }
    }
    for (int i = n - m + 1; i <= n; ++i) {
        w[i] = read();
        f[i][1] = w[i];
        s[i] = 1;
    }
    dfs(1);
    for (int i = m; ~i; i--) {
        if (f[1][i] >= 0) {
            printf("%d\n", i);
            return 0;
        }
    }
}