洛谷 P2515 软件安装 —— tarjan + 树形背包

题目链接：点我啊╭(╯^╰)╮

题目大意：

    每个点有重量、价值、依赖点
    树形背包模型，求体积为 $M$ 的背包的最大值

解题思路：

    注意根据依赖点建图后不是树和森林
    因为会形成环，环上的点要么都选，要么都不选
    因此 $tarjan$ 缩点后跑一下树形背包即可
    这里用了 $dfs$ 序的 $O(nw)$ 算法
    $namo$ 为什么 P2014 [CTSC1997]选课 这道题就是森林而不会形成环呢？
    注意数组越界报wa会多折腾几个小时

#include<bits/stdc++.h>
#define rint register int
#define deb(x) cerr<<#x<<" = "<<(x)<<'\n';
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const int maxn = 2e3 + 5;
int n, m, d[maxn], w[maxn], v[maxn]; 
int nw[maxn], nv[maxn], nw2[maxn], nv2[maxn];
int dfn[maxn], low[maxn], vis[maxn], col[maxn];
int s[maxn], top, tot, numc, nxt[maxn];
int dp[maxn][505], in[maxn];
vector <int> g[maxn];

void tarjan(int u){
	dfn[u] = low[u] = ++tot;
	s[++top] = u;
	vis[u] = 1;
	for(auto v : g[u]){
		if(!dfn[v]){
			tarjan(v);
			low[u] = min(low[u], low[v]);
		} else if(vis[v]) low[u] = min(low[u], dfn[v]);
	}
	if(dfn[u] == low[u]){
		++numc;
		while(s[top+1] ^ u){
			col[s[top]] = numc;
			nw[numc] += w[s[top]]; 
			nv[numc] += v[s[top]]; 
			vis[s[top--]] = 0;
		}
	}
}

void dfs(int u){
	dfn[u] = tot++;
	for(auto v : g[u]) dfs(v);
	nxt[dfn[u]] = tot;
}

signed 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);
	for(int i=1; i<=n; i++) {
		scanf("%d", d+i);
		if(d[i]) g[d[i]].push_back(i);
	}
	for(int i=1; i<=n; i++)
		if(!dfn[i]) tarjan(i);
		
	for(int i=0; i<=n; i++) g[i].clear();
	for(int i=1; i<=n; i++)
		if(col[i] ^ col[d[i]] && d[i]) 
			g[col[d[i]]].push_back(col[i]), in[col[i]]++;
	for(int i=1; i<=numc; i++)
		if(!in[i]) g[0].push_back(i);
		
	dfs(tot = 0);
	memset(dp, 128, sizeof(dp));
	memset(dp[0], 0, sizeof(dp[0]));
	for(int i=1; i<=numc; i++) nw2[dfn[i]] = nw[i];
	for(int i=1; i<=numc; i++) nv2[dfn[i]] = nv[i];
	for(int i=0; i<=numc; i++)
		for(int j=0; j<=m; j++){
			dp[nxt[i]][j] = max(dp[nxt[i]][j], dp[i][j]);
			if(j + nw2[i] > m) continue; 
			dp[i+1][j+nw2[i]] = max(dp[i+1][j+nw2[i]], dp[i][j] + nv2[i]);
		}
	printf("%d\n", dp[numc+1][m]);
}