poj 1155 TELE (树形dp)

题意：

有一个电视台广播节目，广播的网络用一棵树表示，节点1表示电台，叶子结点表示用户，用户愿意付一定

的钱去收看这个节目，从非叶子结点到其他结点需要一定的费用（即从中继点到另一个中继点需要一些钱）

，问最后在不亏本的情况下，最多能使多少人收看到节目。

分析：

1、定义dp[i][j]表示在节点i为根节点的子树下，有j个人收看节点的最大利润。

状态转移方程是：dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[son of i][k])

2、由下向上递推（像数塔一样），每次把从当前点向下可达的点的数量算出作为背包容量。

3、如果用dp[i][j+k]=max(dp[i][j+k],dp[i][j]+dp[son of i][k]这个变形公式(详见代码2)，则需要先将dp[i][j]的值

先记录在tmp[j]数组里。因为更新dp[i][j+k]时， 从小到大的顺序，可能用到dp[i][j]时，它已经是更新后的值了，

这样就会有影响。

AC代码1：

#include<cstdio>
#include<iostream>
#include<cstring>
#define INF 99999999

using namespace std;

struct edge
{
    int to,v;
    int next;
}e[3010];

int dp[3010][3010],num[3010],tot,head[3010],n,m,tmp[3010];

int maxx(int a,int b)
{
    return a>b?a:b;
}

void add(int a,int b,int c)
{
    e[tot].to=b;
    e[tot].v=c;
    e[tot].next=head[a];
    head[a]=tot++;
}

void dfs(int x)
{
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        dfs(v);
        num[x]+=num[v];
        for(int j=num[x];j>=0;j--)
        {
            for(int k=0;k<=j;k++)
            {
                if(dp[x][j-k]!=INF && dp[v][k]!=INF)
                    dp[x][j]=maxx(dp[x][j],dp[x][j-k]+dp[v][k]-e[i].v);
            }
        }
    }
}

int main()
{
    int k,a,b;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof(head));
        tot=0;
        for(int i=1;i<=n-m;i++)
        {
            num[i]=0;
            scanf("%d",&k);
            for(int j=1;j<=k;j++)
            {
                scanf("%d%d",&a,&b);
                add(i,a,b);
            }
        }
        for(int i=1;i<=n;i++)
        {
            //dp[i][0]=0;
            for(int j=1;j<=m;j++)
                dp[i][j]=-INF;
        }
        for(int i=n-m+1;i<=n;i++)
        {
            scanf("%d",&dp[i][1]);
            num[i]=1;
        }
        dfs(1);
        for(int i=m;i>=0;i--)
        {
            if(dp[1][i]>=0)
            {
                printf("%d\n",i);
                break;
            }
        }
    }
    return 0;
}

AC代码2：

#include<cstdio>
#include<iostream>
#include<cstring>
#define INF 99999999

using namespace std;

struct edge
{
    int to,v;
    int next;
}e[3010];

int dp[3010][3010],num[3010],tot,head[3010],n,m,tmp[3010];

int maxx(int a,int b)
{
    return a>b?a:b;
}

void add(int a,int b,int c)
{
    e[tot].to=b;
    e[tot].v=c;
    e[tot].next=head[a];
    head[a]=tot++;
}

void dfs(int x)
{
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        dfs(v);
        for(int j=0;j<=num[x];j++)
            tmp[j]=dp[x][j];
        for(int j=0;j<=num[x];j++)
        {
            for(int k=1;k<=num[v];k++)
            {
                dp[x][j+k]=maxx(dp[x][j+k],tmp[j]+dp[v][k]-e[i].v);
            }
        }
        num[x]+=num[v];
    }
}

int main()
{
    int k,a,b;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof(head));
        tot=0;
        for(int i=1;i<=n-m;i++)
        {
            num[i]=0;
            scanf("%d",&k);
            for(int j=1;j<=k;j++)
            {
                scanf("%d%d",&a,&b);
                add(i,a,b);
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
                dp[i][j]=-INF;
        }
        for(int i=n-m+1;i<=n;i++)
        {
            scanf("%d",&dp[i][1]);
            num[i]=1;
        }
        dfs(1);
        for(int i=m;i>=0;i--)
        {
            if(dp[1][i]>=0)
            {
                printf("%d\n",i);
                break;
            }
        }
    }
    return 0;
}