bzoj 2427: [HAOI2010]软件安装（Trajan+树形DP）

2427: [HAOI2010]软件安装

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1844  Solved: 723
[ Submit][ Status][ Discuss]

Description

现在我们的手头有N个软件，对于一个软件i，它要占用W_i的磁盘空间，它的价值为V_i。我们希望从中选择一些软件安装到一台磁盘容量为M计算机上，使得这些软件的价值尽可能大（即V_i的和最大）。
但是现在有个问题：软件之间存在依赖关系，即软件i只有在安装了软件j（包括软件j的直接或间接依赖）的情况下才能正确工作（软件i依赖软件j)。幸运的是，一个软件最多依赖另外一个软件。如果一个软件不能正常工作，那么它能够发挥的作用为0。
我们现在知道了软件之间的依赖关系：软件i依赖软件D_i。现在请你设计出一种方案，安装价值尽量大的软件。一个软件只被安装一次，如果一个软件没有依赖则D_i=0，这时只要这个软件安装了，它就能正常工作。

Input

第1行：N, M  （0<=N<=100, 0<=M<=500）
      第2行：W₁, W₂, ... W_i, ..., W_n （0<=W_i<=M ）
      第3行：V₁, V₂, ..., V_i, ..., V_n  （0<=Vi<=1000 ）
      第4行：D₁, D₂, ..., D_i, ..., D_n （0<=D_i<=N, D_i≠i ）

Output

一个整数，代表最大价值。

Sample Input

3 10

5 5 6

2 3 4

0 1 1

Sample Output

5

感觉实在没有精力去写一道Trajan+树形DP的题，一眼的思路，代码却很麻烦

不会想去抄个板子吧把这题过了，更不想重新敲Trajan一遍又一遍

在我看来都太累人了

真的要是最后WA了，你都不知道问题出在哪里

DP出问题，建边建错，变量写错……感觉一个都能DEBUG半天

写吧。。要DEBUG的话感觉至少要1个小时，毕竟太菜，就不想写

这题要放在比赛的时候估计是这样的：

我：唉这题好像一个Trajan+树形DP就可以搞定哎，不过好麻烦可能要写很久，貌似榜上过的人也不多

队友：你在说什么？反正我不会

我：好吧，我也不太会，放着吧，先跟榜

比赛结束：？？？？我去是不是中间有道题我说有思路来着？后面就再也没管了，好像真的是这样搞

不要笑太菜就是这样子的

现在感觉搞ACM得要不是dalao，要不就是像我这样的抖M

……

算了还是写吧。。谁叫我真的是抖M

#include<stdio.h>
#include<vector>
#include<stack>
#include<algorithm>
using namespace std;
vector<int> G[505], G2[505];
stack<int> st;
int cnt, Time, vis[505], scc[505], low[505], t[505], cost[505], val[505], c[505], vl[505], dp[505][505], in[505];
void Trajan(int u)
{
	int i, v;
	st.push(u);
	vis[u] = 1;
	t[u] = low[u] = ++Time;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i];
		if(vis[v]==0)
		{
			Trajan(v);
			low[u] = min(low[u], low[v]);
		}
		else if(scc[v]==0)
			low[u] = min(low[u], t[v]);
	}
	if(low[u]==t[u])
	{
		cnt++;
		while(st.empty()==0)
		{
			v = st.top();
			st.pop();
			scc[v] = cnt;
			c[cnt] += cost[v];
			vl[cnt] += val[v];
			if(v==u)
				break;
		}
	}
}
void Sech(int u)
{
	int i, j, k, v;
	for(i=0;i<G2[u].size();i++)
	{
		v = G2[u][i];
		Sech(v);
		for(j=500;j>=0;j--)
		{
			for(k=0;k<=j;k++)
				dp[u][j] = max(dp[u][j], dp[u][j-k]+dp[v][k]);
		}
	}
	for(i=500;i>=0;i--)
	{
		if(i>=c[u])
			dp[u][i] = dp[u][i-c[u]]+vl[u];
		else
			dp[u][i] = 0;
	}
}
int main(void)
{
	int n, V, i, j, x;
	scanf("%d%d", &n, &V);
	for(i=1;i<=n;i++)
		scanf("%d", &cost[i]);
	for(i=1;i<=n;i++)
		scanf("%d", &val[i]);
	for(i=1;i<=n;i++)
	{
		scanf("%d", &x);
		if(x!=0)
			G[x].push_back(i);
	}
	for(i=1;i<=n;i++)
	{
		if(vis[i]==0)
			Trajan(i);
	}
	while(st.empty()==0)
	{
		x = st.top();
		st.pop();
		scc[x] = ++cnt;
		c[cnt] += cost[x];
		vl[cnt] += val[x];
	}
	for(i=1;i<=n;i++)
	{
		for(j=0;j<G[i].size();j++)
		{
			x = G[i][j];
			if(scc[i]!=scc[x])
			{
				in[scc[x]] = 1;
				G2[scc[i]].push_back(scc[x]);
			}
		}
	}
	for(i=1;i<=cnt;i++)
	{
		if(in[i]==0)
			G2[0].push_back(i);
	}
	Sech(0);
	printf("%d\n", dp[0][V]);
	return 0;
}
/*
5 4
1 1 1 1 1
1 2 3 4 5
0 1 2 3 4
*/