poj 1155 树形dp 背包

题意：求最多的用户量。

dp【u】【j】代表节点u用户数为 j 时的花费。

方程：

 dp[ u ] [ j ] = max ( dp [ u ] [ j ] , dp [ u ] [ j - k ] + dp [ v ] [ k ] - w ) ;

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
using namespace std;
const int N=3003;
struct Point
{
    int v,w;
};
vector<Point>V[N];
int dp[N][N],num[N];
void init(int n)
{
    for(int i=0;i<=n;i++)
      V[i].clear();
    memset(dp,-100,sizeof(dp));
    memset(num,0,sizeof(num));
}
void dfs(int u)
{
    dp[u][0] = 0;
    for(int i=0;i<(int)V[u].size();i++)
    {
        int v=V[u][i].v;
        int w=V[u][i].w;
        dfs(v);
        num[u]+=num[v];
        for(int j=num[u];j>=1;j--)
          for(int k=1;j-k>=0&&k<=num[v];k++)
            dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]-w);
    }
}
int main()
{
    //freopen("Input.txt","r",stdin);
    int n,m,i;
    while(~scanf("%d%d",&n,&m))
    {
        int k;
        init(n);
        for(i=1;i<=n-m;i++)
        {
            scanf("%d",&k);
            while(k--)
            {
                int a,b;Point tmp;
                scanf("%d%d",&a,&b);
                tmp.v=a;tmp.w=b;
                V[i].push_back(tmp);
            }
        }
        for(i=n-m+1;i<=n;i++)
        {
            scanf("%d",&k);
            dp[i][1]=k;
            num[i]=1;
        }
        dfs(1);
        for(i=m;i>=1;i--)
        {
            if(dp[1][i]>=0)
            {
                printf("%d\n",i);
                break;
            }
        }
    }
    return 0;
}