poj 1155 TELE 树形dp

/*
大意：某电台要广播一场比赛，该电台网络是由N个网点组成的一棵树，其中M个点为客户端，
其余点为转发站。客户端i愿支付的钱为pay[i],每一条边需要的花费固定，问电台在保证不亏损的情况下，
最多能使多少个客户端接收到信息？广播台所在的节点编号为1
dp[i][j] 表示以i 为根j个叶子的最大值！找dp[1][j]>=0的最大j值即可
走了两遍dfs 才告别了TLE
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=3002;
const int inf=1<<29;
int n,m,dp[maxn][maxn],cnt[maxn],val[maxn],NE,head[maxn];
struct node
{
    int u,v,w,next;
}Edge[maxn<<2];
void addEdge(int u,int v,int w)
{
    Edge[NE].u=u;
    Edge[NE].v=v;
    Edge[NE].w=w;
    Edge[NE].next=head[u];
    head[u]=NE++;
}
void init()
{
    NE=0;
    memset(head,-1,sizeof(head));
    memset(cnt,0,sizeof(cnt));
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            dp[i][j]=-inf;
        }
    }
}
void dfs1(int u,int fa)
{
    int i,j,k,v;
    if(u>=n-m+1)cnt[u]=1;
    for(i=head[u];i!=-1;i=Edge[i].next)
    {
        v=Edge[i].v;
        if(v==fa)continue;
        dfs1(v,u);
        cnt[u]+=cnt[v];
    }
}
void dfs(int u,int fa)
{
    int i,j,k,v;
    dp[u][0]=0;
    if(u>=n-m+1) dp[u][1]=val[u];
    for(i=head[u];i!=-1;i=Edge[i].next)
    {
        int v=Edge[i].v,w=Edge[i].w;
        if(v==fa)continue;
        dfs(v,u);
        for(j=cnt[u];j>=1;j--)
        {
            for(k=1;k<=j;k++)
            {
                dp[u][j]=max(dp[u][j],dp[v][k]+dp[u][j-k]-w);
            }
        }
    }
}
int main()
{
    int i,j,k,v,w;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(i=1;i<=n-m;i++)
        {
            scanf("%d",&k);
            for(j=1;j<=k;j++)
            {
                scanf("%d%d",&v,&w);
                addEdge(i,v,w);
            }
        }
        for(i=n-m+1;i<=n;i++)
        {
            scanf("%d",&val[i]);
        }
        dfs1(1,0);
        dfs(1,-1);
        for(i=m;i>=1;i--)
        if(dp[1][i]>=0)break;
        printf("%d\n",i);
    }
    return 0;
}
/*
9 6
3 2 2 3 2 9 3
2 4 2 5 2
3 6 2 7 2 8 2
4 3 3 3 1 1
*/