cqyz oj | 有线电视网

Description

　　某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构，这棵树的根结点位于足球比赛的现场，

　　树叶为各个用户终端，其他中转站为该树的内部节点。从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的，

　　一场转播的总费用等于传输信号的费用总和。

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

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

Input

　　第一行包含两个用空格隔开的整数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的费用。

　　最后一行依次表示所有用户为观看比赛而准备支付的钱数。

Output

　　仅一行，包含一个整数，表示上述问题所要求的最大用户数。

Sample Input 1

Sample Output 1

Hint

2≤N≤3000

树上的分组背包，和重建道路那个题很像

dp[u][j]表示以u为根选j个点的最大收入，

dp[u][j] = max(dp[u][j-k] + dp[v][k] - edge.w[u][v])

其中v是u的一个儿子，k是在v的子树里选择的叶节点数

注意一下初始化就完事了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 3005, maxm = 3005;

int fir[maxn], ne[maxm], to[maxm], w[maxm], np=0;
void add(int x, int y, int z) {
    ne[++np] = fir[x];
    fir[x] = np;
    to[np] = y;
    w[np] = z;
}

int dp[maxn][maxn];
int cost[maxn];
int n, m;
void data_in() {
    memset(dp, -0x3f, sizeof(dp));
    scanf("%d%d", &n, &m);
    for(int i=1, k, a, c; i<=n-m; i++) {
        scanf("%d", &k);
        while(k--) {
            scanf("%d%d", &a, &c);
            add(i, a, c);
        }
    }
    for(int i=1; i<=m; i++)
        scanf("%d", &cost[n-m+i]);
}

int sz[maxn];
void DFS(int u) {
    dp[u][0] = 0;
    if(u>n-m) {
        sz[u] = 1;
        dp[u][1] = cost[u];
        return;
    }
    sz[u] = 0;
    for(int i = fir[u]; i; i=ne[i]) {
        int v = to[i];
        DFS(v);
        sz[u] += sz[v];
        for(int j = sz[u]; j>=1; j--)
            for(int k = 1; k <= sz[v]; k++)
                if(j-k>=0) dp[u][j] = max(dp[u][j], dp[u][j-k] + dp[v][k] - w[i]);
    }
}

void solve() {
    DFS(1);
    for(int i=m; i>=1; i--) {
        if(dp[1][i] >= 0) {
            printf("%d", i);
            return;
        }
    }
}

int main() {
    data_in();
    solve();
    return 0;
}
/*
5 3
2 2 2 5 3
2 3 2 4 3
3 4 2

*/

View Code