bzoj2427: [HAOI2010]軟體安裝(強聯通+樹形Dp)

Hanks_o發表於2018-03-13

題目傳送門

解法:
有些可能互相依賴變成環。
這時候我們需要縮點。
用強聯通即可。
然後樹形Dp強制選父親才能選兒子即可。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
struct node {int x,y,next;}a[110],e[110];int len,last[110],len1,last1[110];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
void ins1(int x,int y) {len1++;e[len1].x=x;e[len1].y=y;e[len1].next=last1[x];last1[x]=len1;}
int dfn[110],low[110],belong[110],sta[110],cnt,tp,id;bool bo[110];
void dfs(int x) {
    dfn[x]=low[x]=++id;sta[++tp]=x;bo[x]=true;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(dfn[y]==-1) {dfs(y);low[x]=min(low[x],low[y]);}
        else if(bo[y]==true)low[x]=min(low[x],dfn[y]);
    }if(dfn[x]==low[x]) {
        cnt++;int i;
        do {i=sta[tp--];belong[i]=cnt;bo[i]=false;}while(i!=x);
    }
}
int w[110],v[110],W[110],V[110],ru[110],n,m,f[110][510],tot[110];
void Dp(int x) {
    tot[x]=W[x];f[x][W[x]]=V[x];
    for(int k=last1[x];k;k=e[k].next) {
        int y=e[k].y;Dp(y);tot[x]+=tot[y];
        for(int i=m;i>=W[x];i--)for(int j=0;j<=tot[y];j++)
            if(i-j>=W[x])f[x][i]=max(f[x][i],f[y][j]+f[x][i-j]);
    }
    for(int i=m;i>=W[x];i--)f[x][i]=max(f[x][i],V[x]);
}
int main() {    
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=1;i<=n;i++)scanf("%d",&v[i]);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++) {int x;scanf("%d",&x);if(x!=0)ins(x,i);}
    memset(dfn,-1,sizeof(dfn));memset(bo,false,sizeof(bo));
    cnt=id=tp=0;memset(sta,0,sizeof(sta));
    for(int i=1;i<=n;i++)if(dfn[i]==-1)dfs(i);
    memset(W,0,sizeof(W));memset(V,0,sizeof(V));
    for(int i=1;i<=n;i++) {W[belong[i]]+=w[i];V[belong[i]]+=v[i];}
    len1=0;memset(last1,0,sizeof(last1));memset(ru,0,sizeof(ru));
    for(int i=1;i<=len;i++) {
        int x=a[i].x,y=a[i].y;
        if(belong[x]!=belong[y]) {ru[belong[y]]++;ins1(belong[x],belong[y]);}
    }
    for(int i=1;i<=cnt;i++)if(ru[i]==0)ins1(cnt+1,i);
    memset(f,0,sizeof(f));Dp(cnt+1);printf("%d\n",f[cnt+1][m]);
    return 0;
}

相關文章