XCOJ 1102 (树形DP+背包)

#include "cstdio"
#include "iostream"
#include "cstring"
using namespace std;
#define maxn 1200
int c[maxn],w[maxn],dp[maxn][maxn],head[maxn],tol;
int n,W,u;
struct Edge
{
    int to,next;
}e[maxn];
void addedge(int u,int v)
{
    e[tol].to=v;
    e[tol].next=head[u];
    head[u]=tol++;
}
void dfs(int root)
{
    int i=root,cost=c[root];
    if(head[root]==-1) for(int i=cost;i<=W;i++) dp[root][i]=w[root];
    for(int a=head[root];a!=-1;a=e[a].next)
    {
        int t=e[a].to;
        dfs(t);
        for(int j=W; j>=1; j--)
            for(int k=1; k<=j; k++)
                dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[t][k]);
    }
    for(int i=cost;i<=W;i++) dp[root][i]=max(dp[root][i],w[root]);
}
int main()
{
    //freopen("in.txt","r",stdin);
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&W);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&u);
        addedge(u,i);
    }
    for(int i=1;i<=n;i++) scanf("%d",&c[i]);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    dfs(1);
    printf("%d\n",dp[1][W]);

}