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];elsedp[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 41 1 1 1 11 2 3 4 50 1 2 3 4*/