【题解】poj1155 树形DP

题目链接
题意：一个树形网络，编号为1的是广播站，叶子节点为广播接收者，要想使用必须付出费用，中间的其他点是中继站，信号到达每个中继站都会有一定的消费；问在保证广播站收益 不亏本的情况下使用者最多是多少？
分析：dp[i][j]表示第i个点下面有j个使用用户的收益；状态转移方程： dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[i].w); 其中u是v的父节点，j枚举u节点的所有子叶点数，k枚举v节点的所有子叶点数，edge[i].w表示u节点下的使用用户不是0的话要减去的成本 dp全部初始化为-inf，dp[i][0]赋为0；

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int dp[3300][3300];
//dp[i][j]表示第i个点下面有j个使用用户的收益;状态转移方程
//dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[i].w)
//u是v的父节点,j枚举u节点的所有子叶点数,k枚举v节点的所有点数
//edge[i].w表示u节点下的使用用户不是0的话要减去的成本 
struct node
{
    int v,w;
    node(int vv,int ww)//学学 
    {
        v=vv;
        w=ww;
    }
};
vector<node>edge[3300];
int cost[3300],sum[3300],use[3300],m,n;
void dfs(int u)
{
    use[u]=1;//标记已经用过 
    for(int i=0;i<(int)edge[u].size();i++)
    {
        int v=edge[u][i].v;//u的子节点v 
        if(!use[v])//v没有跑过 
        {
            dfs(v);//往下搜 
            sum[u]+=sum[v];//加上子节点用户个数 
            for(int j=sum[u];j>=0;j--)
            for(int k=1;k<=sum[v];k++)
            if(j>=k)
            dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-edge[u][i].w);
        }
    }
    if(u>n-m)//走到潜在用户 
    {
        sum[u]=1;//计数为1 
        dp[u][1]=cost[u];//1个潜在用户的收益 
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    int i,j,k,c;
    while(scanf("%d%d",&n,&m)!=-1)
    {
        for(i=1;i<=n;i++)
        edge[i].clear();
        for(i=1;i<=n-m;i++)
        {
            scanf("%d",&k);
            while(k--)
            {
                scanf("%d%d",&j,&c);//号码，费用 
                edge[i].push_back(node(j,c));
                edge[j].push_back(node(i,c));
            }
        }
        for(i=n-m+1;i<=n;i++)
        scanf("%d",&cost[i]);//支付价格 
        for(i=1;i<=n;i++)
        {
            dp[i][0]=0;
            for(j=1;j<=m;j++)
            dp[i][j]=-INF;
        }
        memset(sum,0,sizeof(sum));
        memset(use,0,sizeof(use));
        dfs(1);//发射机 
        for(i=m;i>=1;i--)
        if(dp[1][i]>=0)
        {
            printf("%d\n",i);
            break;
        }
    }
    return 0;
}