【ARC083E】Bichrome Tree 【树形dp】

题解
Tips：我们会发现无论当前点是黑点还是白点，dp数组都是一样的。所以只用对其中一种情况dp就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1005,inf=0x3f3f3f3f;
int n,u,cnt,head[N],x[N],f[2][N*5],g[N],to[N],nxt[N];
void adde(int u,int v){
    to[++cnt]=v;
    nxt[cnt]=head[u];
    head[u]=cnt;
}
void dfs(int u){
    for(int i=head[u];i;i=nxt[i]){
        dfs(to[i]);
    }
    int sta=0;
    memset(f[sta],63,sizeof(f[sta]));
    f[sta][0]=0;
    for(int i=head[u];i;i=nxt[i]){
        int v=to[i];
        sta^=1;
        memset(f[sta],63,sizeof(f[sta]));
        for(int j=x[u];j>=0;j--){
            if(j-x[v]>=0){
                f[sta][j]=min(f[sta][j],f[sta^1][j-x[v]]+g[v]);
            }
            if(j-g[v]>=0){
                f[sta][j]=min(f[sta][j],f[sta^1][j-g[v]]+x[v]);
            }
        }
    }
    for(int i=0;i<=x[u];i++){
        g[u]=min(g[u],f[sta][i]);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=2;i<=n;i++){
        scanf("%d",&u);
        adde(u,i);
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&x[i]);
    }
    memset(g,63,sizeof(g));
    dfs(1);
    puts(g[1]<inf?"POSSIBLE":"IMPOSSIBLE");
    return 0;
}