summaryrefslogtreecommitdiff
path: root/sources/strongly_connected.py
blob: c81725d37ab36361af6203ca181828a6e21a5f91 (plain)
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
# Uses python3

import sys

sys.setrecursionlimit(200000)


def explore(v, adj, visited, postV, clock):
    visited[v] = True
    for w in adj[v]:
        if visited[w] == False:
            explore(w, adj, visited, postV, clock)
    postV[v] = clock[0]
    clock[0] = clock[0] + 1


def reverseG(adj):
    adjReverse = [[] for _ in range(len(adj))]
    for v in range(len(adj)):
        for w in adj[v]:
            adjReverse[w].append(v)

    return adjReverse


def dfs(adj, visited, postV, clock):
    for v in range(len(adj)):
        if visited[v] == False:
            explore(v, adj, visited, postV, clock)


def exploreWithoutPostV(v, adj, visited):
    visited[v] = True
    for w in adj[v]:
        if visited[w] == False:
            exploreWithoutPostV(w, adj, visited)


def number_of_strongly_connected_components(adj):
    adjReversal = reverseG(adj)
    adjReversalVisited = [False] * len(adj)
    postV = [0] * len(adj)
    clock = [0]

    dfs(adjReversal, adjReversalVisited, postV, clock)
    temp = sorted(range(len(postV)), key=lambda k: postV[k], reverse=True)

    visited = [False] * len(adj)
    result = 0
    for i in temp:
        if visited[i] == False:
            exploreWithoutPostV(i, adj, visited)
            result = result + 1
    return result


if __name__ == '__main__':
    input = sys.stdin.read()
    data = list(map(int, input.split()))
    n, m = data[0:2]
    data = data[2:]
    edges = list(zip(data[0:(2 * m):2], data[1:(2 * m):2]))
    adj = [[] for _ in range(n)]
    for (a, b) in edges:
        adj[a - 1].append(b - 1)
    print(number_of_strongly_connected_components(adj))