[HAOI2010]软件安装[缩点 树形dp]

[BZOJ2427] [luoguP2515]

每个人都至多有一个依赖点 所以有可能成环 先tarjan缩点并处理出环的信息wei va

再将缩点后的图重新连接 将入度为0的点和0点连接 这样就构成了一颗树

由此做树形dp 只有父亲节点选了它的儿子才能选 所以在访问进去时先把wei[u]~m都赋为va[u] 我会说我又因为树形dp卡了半天吗

#include<bits/stdc++.h>
using namespace std;
#define Max(x,y) ((x)>(y)?(x):(y))
#define Min(x,y) ((x)<(y)?(x):(y))
const int N=100+5,M=500+5,inf=0x3f3f3f3f,P=19650827;
int n,m,in[N],w[N],val[N],f[N][M];
template <class t>void rd(t &x){
    x=0;int w=0;char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    x=w?-x:x;
}

int head[N],tot=0;
struct edge{int u,v,nxt;}e[N],E[N];
void add(int u,int v){
	e[++tot]=(edge){u,v,head[u]},head[u]=tot;
}
int idx=0,Bcnt=0,dfn[N],bl[N],low[N],wei[N],va[N];
stack<int>s;bool inst[N];
void tarjan(int u){
	dfn[u]=low[u]=++idx;
	s.push(u),inst[u]=1;
	for(int i=head[u],v;i;i=e[i].nxt){
		v=e[i].v;
		if(!dfn[v]) tarjan(v),low[u]=Min(low[u],low[v]);
		else if(inst[v]&&dfn[v]<low[u]) low[u]=dfn[v];
	}
	if(low[u]==dfn[u]){
		int v;++Bcnt;
		do{
			v=s.top(),s.pop();
			inst[v]=0,bl[v]=Bcnt,wei[Bcnt]+=w[v],va[Bcnt]+=val[v];
		}while(u!=v);
	}
}

int hd[N],tt=0;
void Add(int u,int v){
	E[++tt]=(edge){u,v,hd[u]},hd[u]=tt;
}
void dfs(int u){
	for(int i=wei[u];i<=m;++i) f[u][i]=va[u];
	for(int i=hd[u],v;i;i=E[i].nxt){
		v=E[i].v,dfs(v);
		for(int j=m-wei[u];j>0;--j)
		for(int k=0;k<=j;++k)
		f[u][j+wei[u]]=Max(f[u][j+wei[u]],f[u][j+wei[u]-k]+f[v][k]);
	}
}

int main(){
//	freopen("in2.txt","r",stdin);
	//freopen("xor.out","w",stdout);
	rd(n),rd(m); 
	for(int i=1;i<=n;++i) rd(w[i]);
	for(int i=1;i<=n;++i) rd(val[i]);
	for(int i=1,u;i<=n;++i){
		rd(u);
		if(u) add(u,i);
	}
	for(int i=1;i<=n;++i)
	if(!dfn[i]) tarjan(i);
	for(int i=1;i<=tot;++i)
	if(bl[e[i].u]!=bl[e[i].v]) Add(bl[e[i].u],bl[e[i].v]),++in[bl[e[i].v]];
	for(int i=1;i<=Bcnt;++i) if(!in[i]) Add(0,i);
	memset(f,0,sizeof(f));dfs(0);
	printf("%d",f[0][m]);
	return 0;
}