UVALive ~ 4287 ~ Proving Equivalences （强连通分量+缩点）

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 5;
struct Edge
{
    int from, to;
    Edge (int from, int to): from(from), to(to) {}
};
struct SCC
{
    int n, m;
    int DFN[MAXN], LOW[MAXN], sccno[MAXN], dfs_clock, scc_cnt;
    vector<Edge> edges;
    vector<int> G[MAXN];
    stack<int> S;
    void init(int n)
    {
        this->n = n, m = 0;
        edges.clear();
        for (int i = 0; i <= n; i++) G[i].clear();
    }
    void AddEdge (int from, int to)
    { 
        edges.push_back(Edge(from, to));
        m = edges.size();
        G[from].push_back(m-1);
    }
    void dfs(int u)
    {
        DFN[u] = LOW[u] = ++dfs_clock;
        S.push(u);
        for (int i = 0; i < G[u].size(); i++)
        {
            Edge e = edges[G[u][i]];
            int v = e.to;
            if (!DFN[v])
            {
                dfs(v);
                LOW[u] = min(LOW[u], LOW[v]);
            }
            else if (!sccno[v])
                LOW[u] = min(LOW[u], DFN[v]);
        }
        if (LOW[u] == DFN[u])
        {
            scc_cnt++;
            while (1)
            {
                int x = S.top(); S.pop();
                sccno[x] = scc_cnt;
                if (x == u) break;
            }
        }
    }
    void find_scc()
    {
        dfs_clock = scc_cnt  = 0;
        memset(DFN, 0, sizeof(DFN));
        memset(sccno, 0, sizeof(sccno));
        for (int i = 0; i < n; i++)
            if (!DFN[i]) dfs(i);
    }

}gao;

int n, m, in0[MAXN], out0[MAXN];
int main()
{
    int T; scanf("%d", &T);
    while (T--)
    {
        scanf("%d%d", &n, &m);
        gao.init(n);
        for (int i = 0; i < m; i++)
        {
            int u, v; scanf("%d%d", &u, &v);
            u--, v--;
            gao.AddEdge(u, v);
        }
        gao.find_scc();
        for (int i = 1; i <= gao.scc_cnt; i++) in0[i] = out0[i] = 0;
        for (int i = 0; i < gao.m; i++)
        {
            int u = gao.edges[i].from, v = gao.edges[i].to;
            if (gao.sccno[u] != gao.sccno[v]) 
                in0[gao.sccno[v]]++, out0[gao.sccno[u]]++;
        }
        int cnt1 = 0, cnt2 = 0;
        for (int i = 1; i <= gao.scc_cnt; i++)
        {
            if (!in0[i]) cnt1++;
            if (!out0[i]) cnt2++;
        }
        int ans = max(cnt1, cnt2);
        if (gao.scc_cnt == 1) ans = 0;
        printf("%d\n", ans);
    }
    return 0;
}
/*
2
4 0
3 2
1 2
1 3
*/