洛谷P2515 软件安装（Tarjan+树上dp）

题目
一篇通俗易懂的题解
代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
const int inf=0x3f3f3f3f;
typedef long long ll;
struct Edge
{
    int from,to;
};
int n,m;
vector<Edge>edges;
vector<int>G[maxn];
void add(int from,int to)
{
    edges.push_back({from,to});
    int m=edges.size();
    G[from].push_back(m-1);
}

void init()
{
    for(int i=0;i<=n;i++)
        G[i].clear();
    edges.clear();
}
int dfn[maxn],low[maxn];
bool instack[maxn];
stack<int>s;
int color[maxn];
int cnt=0;
int cntb=0;

void tarjan(int u)
{
    dfn[u]=low[u]=++cnt;
    s.push(u);
    instack[u]=true;
    for(int i=0;i<G[u].size();i++)
    {
        Edge e=edges[G[u][i]];
        int v=e.to;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(instack[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        cntb++;
        int node;
        do
        {
            node=s.top();
            s.pop();
            instack[node]=false;
            color[node]=cntb;
        }while(node!=u);
    }
}
int v[maxn],w[maxn],cost[maxn],val[maxn],f[1010][5050],up[maxn],in[maxn];
void dfs(int u)
{
    for(int i=cost[u];i<=m;i++)
        f[u][i]=val[u];
    for(int i=0;i<G[u].size();i++)
    {
        Edge e=edges[G[u][i]];
        int v=e.to;
        dfs(v);
        for(int j=m-cost[u];j>=0;j--)
        {
            for(int q=0;q<=j;q++)
                f[u][j+cost[u]]=max(f[u][j+cost[u]],f[u][j+cost[u]-q]+f[v][q]);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>w[i];
    for(int i=1;i<=n;i++)
        cin>>v[i];
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        up[i]=x;
        if(x!=0)
            add(x,i);
    }
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }
    init();
    for(int i=1;i<=n;i++)
    {
        val[color[i]]+=v[i];
        cost[color[i]]+=w[i];
        if(color[i]!=color[up[i]]&&up[i]!=0)
        {
            add(color[up[i]],color[i]);
            in[color[i]]++;
        }
    }
    int s=cntb+1;
    for(int i=1;i<=cntb;i++)
    {
        if(!in[i])
            add(s,i);
    }
    dfs(s);
    cout<<f[s][m+cost[s]];


}