ZOJ 3795 Grouping 强连通分量-tarjan

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <stack>
using namespace std;
const int N = 100010;
stack<int> sta;
vector <int> G[N];
vector <int> G4[N];
int vis[N];
int n,m;
int pre[N],lowlink[N],sccno[N],w[N],dfs_clk,scc_cnt;
int dp[N];
void init()
{
    dfs_clk=scc_cnt=0;
    for (int i=0;i<=n;i++){
        G[i].clear();
        pre[i]=0;
        lowlink[i]=0;
        sccno[i]=0;
        vis[i]=0;
        w[i]=0;
        G4[i].clear();
    }
}
void dfs(int u)
{
    pre[u]=lowlink[u]=++dfs_clk;
    sta.push(u);
    for (int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if (!pre[v]){
            dfs(v);
            lowlink[u]=min(lowlink[u],lowlink[v]);
        }
        else if (!sccno[v]){
            lowlink[u]=min(lowlink[u],pre[v]);
        }
    }
    if (lowlink[u]==pre[u]){
        scc_cnt++;
        for (;;){
            int x=sta.top();sta.pop();
            sccno[x]=scc_cnt;
            w[scc_cnt]++;
            if (x==u) break;
        }
    }
}
void tarjan()
{
    for (int i=1;i<=n;i++){
        if (!pre[i]) dfs(i);
    }
}
void calc(int u)
{
    if (dp[u]!=-1) return;
    dp[u]=w[u];
    int now=w[u];
    for (int j=0;j<G4[u].size();j++){
        int v=G4[u][j];
        calc(v);
        dp[u]=max(dp[u],dp[v]+now);
    }
}
void solve()
{
    for (int i=1;i<=n;i++){
        int u=sccno[i];
        for (int j=0;j<G[i].size();j++){
            int v=sccno[G[i][j]];
            if (u!=v) G4[u].push_back(v);
        }
    }
    for (int i=1;i<=scc_cnt;i++){
        dp[i]=-1;
    }
    for (int i=1;i<=scc_cnt;i++){
        calc(i);
    }
    int ans=0;
    for (int i=1;i<=scc_cnt;i++){
        ans=max(ans,dp[i]);
    }
    printf("%d\n",ans);
}
int main()
{
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        int a,b;
        while (m--)
        {
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
        }
        tarjan();
        solve();
    }
    return 0;
}