P2515 [HAOI2010]软件安装

树形背包

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<vector>
#define MAXN 110
#define MAXM 510
#define pii pair<int,int>
#define rint register int
#define mp make_pair
#define pb push_back
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if('-'==ch)f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;    
}
int tot,m;
int C[MAXN],V[MAXN];
vector<int> G[MAXN];
int dfn[MAXN],low[MAXN],sta[MAXN],b[MAXN],top,idx;
int c[MAXN],v[MAXN],cmp[MAXN],n;
int fst[MAXN],nxt[MAXN],from[MAXN],to[MAXN],cnt,pin[MAXN];
void add(int x,int y){
    nxt[++cnt]=fst[x],fst[x]=cnt,from[cnt]=x,to[cnt]=y;    
    pin[y]++;
}
void tarjan(int x){
    dfn[x]=low[x]=(++idx);
    sta[++top]=x,b[x]=1;
    int y;
    for(rint i=0;i<G[x].size();i++){
        y=G[x][i];
        if(!dfn[y]){
            tarjan(y);    
            low[x]=min(low[x],low[y]);
        }
        else if(b[y]){
            low[x]=min(low[x],dfn[y]);
        }
    }
    if(dfn[x]==low[x]){
        n++;
        while(sta[top+1]!=x){
            c[n]+=C[sta[top]];
            v[n]+=V[sta[top]];
            cmp[sta[top]]=n;
            b[sta[top]]=0;
            top--;    
        }
    }
}
int f[MAXN][MAXM],dep[MAXN];
int H[MAXM],H1[MAXM],H2[MAXM];
void work(int x,int vl){
    if(vl<c[x]){f[x][vl]=0;return;}
    vl-=c[x];
    memset(H,0,sizeof(H));
    memset(H1,0,sizeof(H1));
    int y,r=0;
    for(rint e=fst[x];e;e=nxt[e]){
        y=to[e];
        for(rint i=0;i<=vl;i++){
            H[i]=max(H[i],H1[i]+f[y][vl-i]);
            r=max(r,H[i]);
        }
        memcpy(H1,H,sizeof(H));
    }
    f[x][vl+c[x]]=r+v[x];
}
void dp(int x){
    b[x]=1;
    int y;
    for(rint e=fst[x];e;e=nxt[e]){
        y=to[e];
        if(!b[y]){
            dp(y);
            for(rint j=m-c[x];j>=0;j--){
                for(rint k=0;k<=j;k++){
                    f[x][j]=max(f[x][j],f[x][k]+f[y][j-k]);
                }
            }
        }
    }
    for(rint j=m;j>=0;j--){
        if(j>=c[x]){
            f[x][j]=f[x][j-c[x]]+v[x];
        }
        else{
            f[x][j]=0;
        }
    }
}
int main()
{
  //  freopen("data.in","r",stdin);
    tot=read();m=read();
    for(rint i=1;i<=tot;i++)C[i]=read();
    for(rint i=1;i<=tot;i++)V[i]=read();
    int tmp;
    for(rint i=1;i<=tot;i++){
        tmp=read();
        if(tmp)G[tmp].pb(i);
    }
    for(rint i=1;i<=tot;i++){
        if(!dfn[i])tarjan(i);    
    }
    for(rint i=1;i<=tot;i++){
        for(rint j=0;j<G[i].size();j++){
            int x=i,y=G[i][j];
            if(cmp[x]!=cmp[y]){
                add(cmp[x],cmp[y]);
            }
        }
    }
    for(rint i=1;i<=n;i++){
        if(!pin[i]){
            add(0,i);    
        }
    }
//    for(rint i=1;i<=cnt;i++){
//        printf("%d %d
",from[i],to[i]);    
//    }
    memset(b,0,sizeof(b));
    dp(0);
    printf("%d
",f[0][m]);
    return 0;    
}