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

思路:把选取子树在不同用户数下获取的利润作为价值,在树上进行背包,不断更新父节点的信息
背包:

for (int i = 0; i <= cnt[u]; i++)t[i]=dp[u][i];//滚动数组,将上一次的结果存下来
for (int j = 1; j <= cnt[u]; j++) {
   for (int k = 1; k <= min(j,cnt[v]); k++) {
      dp[u][j] = max(dp[u][j], t[j-k] + dp[v][k]-cost[v]);
    }
 }

代码:

#include<iostream>
#include<string>
#include<stack>
#include<stdio.h>
#include<queue>
#include<string.h>
#include<map>
#include<unordered_map>
#include<vector>
#include<iomanip>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
const int maxn = 3001;
const int maxm = 3001;
inline int read()
{
    int f = 1, num = 0;
    char ch = getchar();
    while (0 == isdigit(ch)) { if (ch == '-')f = -1; ch = getchar(); }
    while (0 != isdigit(ch)) num = (num << 1) + (num << 3) + ch - '0', ch = getchar();
    return num * f;
}
struct edge {
    int to, nxt;
}e[maxm];
int hd[maxn], tot;
void add(int f, int t) {
    e[++tot].to = t;
    e[tot].nxt = hd[f];
    hd[f] = tot;
}
int cost[maxn],cnt[maxn],money[maxn];
ll dp[maxn][maxn];
int n, m;
int t[maxn];
void dfs(int u) {
    if (hd[u] == 0) {
        dp[u][1] = money[u];
        cnt[u] = 1;
        return;
    }
    for (int i = hd[u]; i; i = e[i].nxt) {
        int v = e[i].to;
        dfs(v);
        cnt[u] += cnt[v];
        for (int i = 0; i <= cnt[u]; i++)t[i]=dp[u][i];//滚动数组,将上一次的结果存下来
        for (int j = 1; j <= cnt[u]; j++) {
            for (int k = 1; k <= min(j,cnt[v]); k++) {
                dp[u][j] = max(dp[u][j], t[j-k] + dp[v][k]-cost[v]);
            }
        }
    }
}
int main() {
    //freopen("test.txt", "r", stdin);
    n = read(), m = read();
    for (int i = 1; i <= n - m; i++) {
        int k = read();
        while (k--) {
            int v = read(), w = read();
            add(i, v);
            cost[v] = w;
        }
    }
    for (int i = n - m + 1; i <= n; i++) {
        money[i] = read();
    }
    for (int i = 1; i <= n; i++) {
        dp[i][0] = 0;
        for (int j = 1; j <= n; j++) {
            dp[i][j] = -inf;
        }
    }
    dfs(1);
    int res=0;
    for (int i = 1; i <= m; i++) {
        if (dp[1][i] >= 0)res = i;
    }
    cout << res << endl;
    return 0;
}