Merge branch 'master' of https://github.com/lucashc/programmingCheatSheet

lucashc · lucashc · commit ed6e2a4d407f · 2019-09-17T09:02:14.000+02:00
diff --git a/algorithms/topological_sort.py b/algorithms/topological_sort.py
@@ -0,0 +1,22 @@
+def topological_sort(graph):
+    from collections import deque
+    indeg = [0] * len(graph)
+    result = []
+    q = deque()
+    for ns in graph:
+        for n in ns:
+            indeg[n] += 1
+
+    for i in range(len(graph)):
+        if indeg[i] == 0:
+            q.appendleft(i)
+
+    while q:
+        n = q.pop()
+        result.append(n)
+        for i in graph[n]:
+            indeg[i] -= 1
+            if indeg[i] == 0:
+                q.appendleft(i)
+    return result
+
