HAOI2010 软件安装 强连通分量+树上背包

题意：有n个软件，有的软件依赖于某些软件，求最大空间为m时的最大收益。
题解：容易发现有些软件互相依赖组成了一个环。此时这些依赖关系组成的基环树中，环没有出边。所以把环缩成点。此时，森林中除了缩过点的树以外，根节点都有一条来自虚拟节点0的入边。因此，我们从0点向每个缩过点的树连边，再做一次树上背包就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 101;
const int MAXM = 501;

struct Edge{
	int fr, to;
}e[MAXN];

int n, m, w[MAXN], v[MAXN];
int fir[MAXN], nxt[MAXN << 1], to[MAXN << 1], cnt;
int dfn[MAXN], low[MAXN], tim;
int sta[MAXN], vis[MAXN], top;
int scc[MAXN], W[MAXN], V[MAXN], size[MAXN], cont;
int dp[MAXN][MAXM];

inline int read(){
	int k = 0, f = 1; char ch = getchar();
	while(ch < '0' || ch > '9'){if(ch == '-') f = -1; ch = getchar();}
	while(ch >= '0' && ch <= '9'){k = k*10 + ch - '0'; ch = getchar();}
	return k * f;
}

inline void add_edge(int a, int b){
	to[cnt] = b;
	nxt[cnt] = fir[a];
	fir[a] = cnt++;
}

void Tarjan(int u){
	dfn[u] = low[u] = ++tim;
	sta[++top] = u, vis[u] = true;
	for(int i = fir[u]; i != -1; i = nxt[i]){
		int v = to[i];
		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]){
		cont++;
		do{
			scc[sta[top]] = cont;
			W[cont] += w[sta[top]];
			V[cont] += v[sta[top]];
			size[cont]++;
			vis[sta[top--]] = false;
		}while(sta[top + 1] != u);
	}
}

void dfs(int u){
	dp[u][0] = 0;
	for(int i = fir[u]; i != -1; i = nxt[i]){
		int v = to[i];
		dfs(v);
		for(int t = m; t >= 0; t--){
			for(int j = t; j >= 0; j--){
				dp[u][t] = max(dp[u][t], dp[u][t - j] + dp[v][j]); //以v的数据更新u 
			}
		}
	}
	if(u){ //u不为0时，u本身要占用一定体积，获得一定收益 
		for(int t = m; t >= 0; t--){
			if(t - W[u] >= 0) dp[u][t] = dp[u][t - W[u]] + V[u];
			else dp[u][t] = 0;
		}
	}
}

int main(){
	freopen("in.txt", "r", stdin);
	memset(fir, -1, sizeof(fir));
	n = read(), m = read();
	
	for(int i = 1; i <= n; i++)
		w[i] = read();
	for(int i = 1; i <= n; i++)
		v[i] = read();
	for(int i = 1; i <= n; i++){
		e[i].fr = read(), e[i].to = i;
		add_edge(e[i].fr, e[i].to);
	}
	
	for(int i = 1; i <= n; i++)
		if(!dfn[i]) Tarjan(i);
	
	memset(fir, -1, sizeof(fir)); cnt = 0;
	for(int i = 1; i <= n; i++)
		if(scc[e[i].fr] != scc[e[i].to])
			add_edge(scc[e[i].fr], scc[e[i].to]);
	for(int i = 1; i <= cont; i++)
		if(size[i] > 1) add_edge(0, i); //size大于1的点被缩过 
	
	dfs(0);
	printf("%d", dp[0][m]);
	
	return 0;
}