POJ 1155 TELE 树形背包

题意：在一个有向树形网络中，内部节点是信息发射台，叶子节点是收听观众。根节点向儿子节点发送信息需要不同的费用。观众收听对应的信息会支付不同的报酬。现在公司想，在收益非负的情况下，让最多的听众收听到比赛信息，问最多可以收听多少听众。
思路：树形DP。
首先，我们可以注意到，题目中给出了N的范围，这启发我们要以N的范围作为DP的下标。
定义状态dp[u][j]为以u为根的子树，让j个听众收听广播所得到的最大收益。
同时利用滚动背包的思想，有：
对于u的儿子v，我们有两种选择：
1.不向儿子v发送信息，这样既没有收益也没有花费。
2.向儿子v发送消息，这样，我们就可以得到以v为根的子树的收益。
最大的收益就是上面两种情况的最大值。
即

$$dp[u][j] = max(dp[u][j], dp[u][j-k] + dp[v][k] - cost_{u,v}) v \in son(u), 1 \le k \le j$$
最后就是找出最大的j，使 $dp[1][j] \ge 0$
边界条件：
$$dp[u][0] = 0$$
注意：在求解的过程中，需要限定j的范围，而不能直接用 $1 \le j \le M$。因为，对于节点u，他能发送到的最多的用户，就是以u为根的子树的叶子节点的个数，所以j的循环范围就是 $1 \le j \le num[u]$
时间复杂度为： $\Omega(n^3)$
代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAX = 3010;

struct edge{
    int to,w;
    edge(){}
    edge(int t, int ww):to(t),w(ww){}
} edges[MAX<<1];

int N,M;
int nxt[MAX<<1],head[MAX],tot;
int dp[MAX][MAX],num[MAX];


void init()
{
    memset(head,-1,sizeof(head));
    memset(dp,0xcf,sizeof(dp));
    tot = 0;
    for(int i = 1; i <= N; ++i)
        dp[i][0] = 0;
}

void addedge(int u, int v, int w)
{
    edges[tot] = edge(v,w);
    nxt[tot] = head[u];
    head[u] = tot++;
}

int dfs(int u, int p)
{
    for(int i = head[u]; ~i; i = nxt[i]){
        edge & e = edges[i];
        if(e.to == p) continue;
        num[u] += dfs(e.to,u);
        for(int j = num[u]; j >= 0; --j)
            for(int k = min(num[e.to],j); k >= 1; --k)
                dp[u][j] = max(dp[u][j],dp[u][j-k] + dp[e.to][k] - e.w);
    }
    return num[u];
}

int main(void)
{
    //freopen("input.txt","r",stdin);
    scanf("%d%d",&N,&M);
    init();
    for(int i = 1; i <= N - M; ++i){
        num[i] = 0;
        int k,a,c;
        scanf("%d",&k);
        for(int j = 1; j <= k; ++j){
            scanf("%d%d",&a,&c);
            addedge(i,a,c);
            addedge(a,i,c);
        }
    }
    for(int i = N - M + 1; i <= N; ++i)
        scanf("%d",&dp[i][1]),num[i] = 1;
    dfs(1,0);
    for(int i = M; i >= 0; --i){
        if(dp[1][i] >= 0){
            printf("%d\n",i);
            break;
        }
    }
    return 0;
}