poj1155 TELE

树形背包

先解释一下题意：一个电视台转播比赛，电视网络类似于树，树的节点为中转站或者用户。

输入：N M N表示转发站和用户总数，M为用户数
以下N-M行，第i行第一个K，表示转发站i和K个（转发站或用户）相连, 其后第k对数val,cost表示，第i个转发站到val有边，费用cost。
最后一行M个数表示每个用户愿意负的钱。

输出：不亏本前提下，可以收到节目最多的用户数。
（如果某个用户要收到节目（叶子结点），那么电视台到该用户的路径节点的费用都要付）。

节点的编号为1~N，其中1为总站，2到(N-M)为中转站，(N-M+1)到N为用户。节点到节点间转移信号需要给定的费用，且每个用户愿意出给定的钱付费。在电视台不亏本的前提下，求最多可以让多少用户看到比赛。
下图最多5个用户，N=9,M=6。

来分析一下算法：由于求的是最多多少用户，那么我们可以把用户个数当成一个状态。dp[i][j]代表i节点为根节点的子树j个用户的时候最大剩余费用。则状态转移方程：

dp[i][j] = max(dp[i][j], dp[i][k]+dp[son][j-k]-w[i][son]);

注意两点，第一点是上面式子中的dp[i][k]必须先用一个tem[MAX]数组提取出来，因为在计算的过程中会相互影响。第二点是价值可能是负值，所以dp初始化的时候要初始化为负的最大值。

再解释一下各个数组代表含义，dp就是记录状态的二维数组，num数组是记录以当前节点为根节点的子树最多可以连多少个用户，tree是模拟树的数组，now为这个点的编号，val为这个点到它的父节点所需要的cost，next是他的父节点。
来看代码

#include<algorithm>
#include<cstdio>
#include<cstring>
#include<iostream>
#define INF 100000007
using namespace std;
struct node
{
    int now,val,next;
} tree[9005];
int n,m,len = 0;
int num[3005],head[3005],dp[3005][3005],tem[3005];
void add(int i,int x,int y)
{
    tree[len].now = x;
    tree[len].val = y;
    tree[len].next = head[i];
    head[i] = len++;
}
void dfs(int root)
{
    int i,j,k,p;
    for(i = head[root]; i!=-1; i=tree[i].next)
    {
        p = tree[i].now;
        dfs(p);
        for(j = 0; j<=num[root]; j++)
            tem[j] = dp[root][j];
        for(j = 0; j<=num[root]; j++)
        {
            for(k = 1; k<=num[p]; k++)
                dp[root][k+j] = max(dp[root][j+k],tem[j]+dp[p][k]-tree[i].val);
        }
        num[root]+=num[p];
    }
}
int main()
{
    int i,j,k,val,cost;
    while(~scanf("%d%d",&n,&m))
    {
        memset(head,-1,sizeof(head));
        for(i = 1; i<=n-m; i++)
        {
            scanf("%d",&k);
            num[i] = 0;
            for(j = 1; j<=k; j++)
            {
                scanf("%d%d",&val,&cost);
                add(i,val,cost);
            }
        }
        for(i = 1; i<=n; i++)
        {
            for(j = 1; j<=m; j++)
                dp[i][j] = -INF;
        }
        for(i = n-m+1; i<=n; i++)
        {
            num[i] = 1;
            scanf("%d",&dp[i][1]);
        }
        dfs(1);
        for(i = m; i>=0; i--)
        {
            if(dp[1][i]>=0)
            {
                printf("%d\n",i);
                break;
            }
        }
    } 
    return 0;
}