ARC 083 E Bichrome Tree - dp

考虑一个朴素dp，设dp[x][i][j][k]表示考虑x这个节点的前i个儿子和x同色的权和是j另一种权和是k是否可行，转移显然，最后只要j<=Xx即可。显然j固定的时候k越小越好，因此对k做背包即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<climits>
#define N 1010
#define MXV 5010
#define INF (INT_MAX/2-10)
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
struct edges{
    int to,pre;
}e[N];int f[N],dp[MXV],X[N],h[N],etop;
inline int add_edge(int u,int v)
{
    return e[++etop].to=v,e[etop].pre=h[u],h[u]=etop;
}
int dfs(int x)
{
    for(int i=h[x];i;i=e[i].pre)
        if(!dfs(e[i].to)) return 0;
    for(int i=1;i<=X[x];i++) dp[i]=INF;
    for(int i=h[x]+(dp[0]=0);i;i=e[i].pre)
    {
        int a=X[e[i].to],b=f[e[i].to];
//      debug(x)sp,debug(e[i].to)sp,debug(a)sp,debug(b)ln;
        for(int j=X[x],t;j>=min(a,b);j--)
        {
            t=dp[j],dp[j]=INF;
            if(j>=a) dp[j]=min(dp[j],(a?dp[j-a]:t)+b);
            if(j>=b) dp[j]=min(dp[j],(b?dp[j-b]:t)+a);
        }
        for(int j=0;j<min(a,b);j++) dp[j]=INF;
    }
    for(int i=0;i<=X[x];i++) f[x]=min(f[x],dp[i]);
//  debug(x)sp,debug(f[x])ln;
    return f[x]<INF;
}
int main()
{
    int n;scanf("%d",&n);
    for(int i=2,x;i<=n;i++) scanf("%d",&x),add_edge(x,i);
    for(int i=1;i<=n;i++) scanf("%d",&X[i]),f[i]=INF;
    return !printf("%s\n",dfs(1)?"POSSIBLE":"IMPOSSIBLE");
}