luogu P1273 有线电视网

题目描述

题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，树叶为各个用户终端，其他中转站为该树的内部节点。
从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛，有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入输出格式
输入格式：
输入文件的第一行包含两个用空格隔开的整数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

题解

• 这道题显然是一道树形$dp$题，根据题意我们的得到如下状态：$f[i][j][k]$ 表示在以$i$为根的子树中选取了前$j$棵子树用户人数为$k$时所得到的最大收益。（像不像背包$qaq$）
• 那么既然这个状态非常想一个背包，那么就可以用滚动数组来优化，不然很可能会$MLE$
• 状态转移方程就很显然了$$f[u][i]=max(f[u][i],f[u][i-j]+f[v][j]-edge[k])$$

code

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e3 + 100;

template <typename T>
inline void read(T &s) {
    s = 0;
    T w = 1, ch = getchar();
    while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
    while (isdigit(ch))  { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
    s *= w;
}

inline int max(int aa, int bb) { return aa > bb ? aa : bb; }
inline int min(int aa, int bb) { return aa < bb ? aa : bb; }

int n, m, id, tot;
int fa[maxn], a[maxn], f[maxn][maxn], lin[maxn], nex[maxn], edge[maxn], ver[maxn];

inline void add(int from, int to, int dis) {
    ver[++tot] = to;
    nex[tot] = lin[from];
    edge[tot] = dis;
    lin[from] = tot;
}

int dfs(int u) {
    if (u > (n - m)) {
        f[u][1] = a[u];
        return 1;
    }
    int ans = 0, t;
    for (int k = lin[u]; k; k = nex[k]) {
        int v = ver[k];
        t = dfs(v); ans += t;
        for (int i = ans; i >= 0; --i) {
            for (int j = 1; j <= t; ++j) {
                if (i >= j) {
                    f[u][i] = max(f[u][i], f[u][i - j] + f[v][j] - edge[k]);
                }
            }
        }
    }
    return ans;
}

int main() {
    memset(f, 0xcf, sizeof(f));
    read(n), read(m);
    for (int i = 1, k; i <= (n - m); ++i) {
        read(k);
        for (int j = 1; j <= k; ++j) {
            int now, val;
            read(now), read(val);
            add(i, now, val);
            // dis[i][now] = val;
        }
    }
    for (int i = (n - m + 1); i <= n; ++i) read(a[i]);
    for (int i = 1; i <= n; ++i) f[i][0] = 0;
    dfs(1);
    for (int i = 1; i <= n; ++i) 
        if (f[1][i] >= 0) id = max(id, i);
    printf("%d\n", id);
    return 0;
}