-
Notifications
You must be signed in to change notification settings - Fork 20
/
Copy pathkosaraju.cpp
70 lines (70 loc) · 1.58 KB
/
kosaraju.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <bits/stdc++.h>
void dfs(int node, unordered_map<int, bool> &vis, stack<int> &s, unordered_map<int, list<int>> &adj)
{
vis[node] = true;
for (auto nbr : adj[node])
{
if (!vis[nbr])
{
dfs(nbr, vis, s, adj);
}
}
// topo logic
s.push(node);
}
void revDfs(int node, unordered_map<int, bool> &vis, unordered_map<int, list<int>> &adj)
{
vis[node] = true;
for (auto nbr : adj[node])
{
if (!vis[nbr])
{
revDfs(nbr, vis, adj);
}
}
}
int stronglyConnectedComponents(int v, vector<vector<int>> &edges)
{
// Write your code here.
// create an adj list
unordered_map<int, list<int>> adj;
for (int i = 0; i < edges.size(); i++)
{
int u = edges[i][0];
int v = edges[i][1];
adj[u].push_back(v);
}
// topo sort
stack<int> s;
unordered_map<int, bool> vis;
for (int i = 0; i < v; i++)
{
if (!vis[i])
{
dfs(i, vis, s, adj);
}
}
// create a transpose graph
unordered_map<int, list<int>> transpose;
for (int i = 0; i < v; i++)
{
vis[i] = 0;
for (auto nbr : adj[i])
{
transpose[nbr].push_back(i);
}
}
// dfs call using above ordering
int count = 0;
while (!s.empty())
{
int top = s.top();
s.pop();
if (!vis[top])
{
count++;
revDfs(top, vis, transpose);
}
}
return count;
}