poj1155(树形dp)

链接：点击打开链接

题意：有1个电视台，N-M-1个中转站，M个用户，电视台可以通过一些线路将信号传输到用户手中，给出每条线路的费用和每个用户可以支付的金额，输出在电视台不会产生亏损的同时能够给多少用户传输信号

代码：

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int SIZE=3005;
const int INF=0x3f3f3f3f;
struct node{
    int to,cost;
};
vector<node> G[SIZE];
int dp[SIZE][SIZE],num[SIZE],used[SIZE];        //dp[i][j]表示第i个节点传输j个用
void dfs(int s){                                //产生的饿利润
    int i,j,k,tmp;
    used[s]=1;
    for(i=0;i<G[s].size();i++){
        tmp=G[s][i].to;
        if(G[tmp].size()&&!used[tmp]){
        dfs(tmp);
        num[s]+=num[tmp];                       //求出每个节点所有子树
        for(j=num[s]-1;j>=0;j--)                //的节点和，包括自己
        for(k=1;k<=j;k++)
        dp[s][j]=max(dp[s][j],dp[s][j-k]+dp[tmp][k]-G[s][i].cost);
        }
    }
}                                               //在树上进行0-1背包
int main(){
    int a,b,i,j,k,n,m,sign;
    while(scanf("%d%d",&n,&m)!=EOF){
        for(i=0;i<=n;i++){
            G[i].clear();
            for(j=0;j<=m;j++){
                if(j==0)
                dp[i][j]=0;
                else
                dp[i][j]=-INF;
            }
        }
        fill(num,num+n+1,1);
        memset(used,0,sizeof(used));
        for(i=1;i<=n-m;i++){                    
            scanf("%d",&k);
            while(k--){
                scanf("%d%d",&a,&b);
                G[i].push_back((node){a,b});
                G[a].push_back((node){i,b});
            }
        }
        for(i=n-m+1;i<=n;i++)
        scanf("%d",&dp[i][1]);                  //输入用户的金额
        dfs(1);
        sign=0;
        for(i=m;i>=0;i--)                       //从后开始盈利的就直接输出
        if(dp[1][i]>=0)
        break;
        printf("%d\n",i);
    }
    return 0;
}