7.9 树上背包3

7.9

树上背包3

给你一个 n(1≤n≤1000)n(1≤n≤1000) 个点的有根树，其中1号点为根。每个点有一个权值 $a_i(1\le a_i\le 10^5)和重量w_i(1\le w_i\le 10^4)$

你需要选择一个重量恰好 kk 的包含根的连通块，最大的权值和。对所有的k=0,1,2,…,$m_k$=0,1,2,…,m输出答案，其中 1≤m≤1041≤m≤104 。如果不存在，输出0。

从后往前做，令f[i][j]表示考虑DFS序中[i,n]这一段的节点，选的重量和不超过j的点集的最大权值和。

转移为f[i][j]=max(f[i][j], f[r[i]+1][j]+w[id[i]]),因为dfs序中r[id[i]]+1正好对应之后的那一个位置

const int N = 2010, M = 20100, INF = 0x3f3f3f3f;
int n,m,k,w[N],a[N],l[N],r[N],idx,id[N];
int f[N][M];
vector<int> edge[N];

void dfs(int u){
	l[++idx]=u,id[idx]=u;
	for(auto v:edge[u]){
		dfs(v);
	}
	r[u]=idx;
}

void solve(){
	n=read(),m=read();
	rep(i,2,n){
		int x=read();
		edge[x].push_back(i);
	}
	rep(i,1,n) a[i]=read();
	rep(i,1,n) w[i]=read();
	dfs(1); 
	for(int i=1;i<=m;++i) f[n+1][i]=-INF;
	for(int i=n;i>=1;--i){
		int u=id[i];
		for(int j=0;j<=m;++j){
			f[i][j]=f[r[u]+1][j];
			if(j>=w[u])
				f[i][j]=max(f[i][j], f[i+1][j-w[u]]+a[u]);
		}
	}
	int ans=0;
	for(int i=0;i<=m;++i)
		if(f[1][i]>=0)
			print(f[1][i]);
		else print(0);
}