【bzoj2427】【软件安装】tarjan缩点+树形依赖背包

#include<stack>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=205;
const int M=1005;

stack<int> s;
int n,m,ww
,vv
;
int/*adde1*/ head1
,end1[N*2],nxt1[N*2],hh1=0;
int/*adde2*/ head2
,end2[N*2],nxt2[N*2],hh2=0;
int/*tarjan*/ dfn
,low
,idc=0,id
,cnt=0;bool exi
;
int cos
,val
,in
;
int/*dfs*/ dp
[M];

void adde1(int a,int b){
hh1++;
end1[hh1]=b;
nxt1[hh1]=head1[a];
head1[a]=hh1;
}
void adde2(int a,int b){
hh2++;
end2[hh2]=b;
nxt2[hh2]=head2[a];
head2[a]=hh2;
}
void tarjan(int u){
dfn[u]=low[u]=++idc;
s.push(u);
exi[u]=1;
for(int i=head1[u];i;i=nxt1[i]){
int v=end1[i];
if(!dfn[v]){
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(exi[v]) low[u]=min(low[u],low[v]);
}
if(low[u]==dfn[u]){
cnt++;
int tmp;
while(s.top()!=u&&(!s.empty())){
tmp=s.top();s.pop();
id[tmp]=cnt;
exi[tmp]=0;
}
if(s.top()==u){
s.pop();
id[u]=cnt;
exi[u]=0;//最后在处理u时忘了。。
}
}
}
void dfs(int u){//printf("%d ",u);
dp[u][0]=0;
for(int i=head2[u];i;i=nxt2[i]){
int v=end2[i];
dfs(v);
for(int k=m-cos[u];k>=0;k--){//用了两次i，就导致了混乱
for(int j=k;j>=0;j--){
dp[u][k]=max(dp[u][k],dp[u][k-j]+dp[v][j]);
}
}
}
for(int j=m;j>=0;j--){
if(j-cos[u]>=0) dp[u][j]=dp[u][j-cos[u]]+val[u];
else dp[u][j]=0;
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&ww[i]);
for(int i=1;i<=n;i++) scanf("%d",&vv[i]);
int d;
for(int i=1;i<=n;i++){
scanf("%d",&d);
if(d) adde1(d,i);
}
while(!s.empty()) s.pop();
for(int i=1;i<=n;i++) if(!dfn[i]) tarjan(i);
for(int u=1;u<=n;u++){
cos[id[u]]+=ww[u],val[id[u]]+=vv[u];
for(int i=head1[u];i;i=nxt1[i]){
int v=end1[i];
if(id[u]!=id[v]) adde2(id[u],id[v]),in[id[v]]++;
}
}
for(int i=1;i<=cnt;i++) if(!in[i]) adde2(0,i);
memset(dp,-27,sizeof(dp));
dfs(0);
int ans=0;
for(int i=1;i<=m;i++) ans=max(ans,dp[0][i]);
printf("%d\n",ans);
return 0;
}