diff --git a/Graph/tarjan_scc.cpp b/Graph/tarjan_scc.cpp index a812aba..7f89859 100644 --- a/Graph/tarjan_scc.cpp +++ b/Graph/tarjan_scc.cpp @@ -1,25 +1,24 @@ -int cnt, scc_num, scc[MN], in[MN], low[MN]; -stack s; -bitset ins; - -void tarjan(int u) { - low[u] = in[u] = cnt++; - s.push(u); - ins[u] = 1; - for (int v : G[u]) { - if (in[v] == -1) tarjan(v), low[u] = min(low[u], low[v]); - else if (ins[v]) low[u] = min(low[u], in[v]); - } - if (low[u] == in[u]) { - while (1) { - int x = s.top(); s.pop(); - scc[x] = scc_num, ins[x] = 0; - if (x == u) break; - } - ++scc_num; - } +struct tarjan { + int cnt, scc_num, scc[MN], in[MN], low[MN]; + stack s; + bitset ins; + void tarjan(vector * G, int u) { + low[u] = in[u] = cnt++, ins[u] = 1; s.push(u); + for (int v : G[u]) { + if (in[v] == -1) tarjan(G, v), low[u] = min(low[u], low[v]); + else if (ins[v]) low[u] = min(low[u], in[v]); + } + if (low[u] == in[u]) { + while (1) { + int x = s.top(); s.pop(); + scc[x] = scc_num, ins[x] = 0; + if (x == u) break; + } + ++scc_num; + } + } + void find_scc(vector * G) { + memset(scc, -1, sizeof scc), memset(in, -1, sizeof in); + for (int u = 1; u <= N; ++u) if (scc[u] == -1) tarjan(G, u); + } } - - -memset(scc, -1, sizeof scc), memset(in, -1, sizeof in); -for (int u = 1; u <= N; ++u) if (scc[u] == -1) tarjan(u);