Update parallel-courses-iii.py

kamyu104 · web-flow · commit 411ea8e656cc · 2021-10-24T18:13:29.000+08:00
diff --git a/Python/parallel-courses-iii.py b/Python/parallel-courses-iii.py
@@ -14,13 +14,8 @@ def minimumTime(self, n, relations, time):
         for prev, nxt in relations:
             adj[prev-1].append(nxt-1)
             in_degree[nxt-1] += 1
-        q = []
-        dist = [0]*n
-        for u in xrange(n):
-            if in_degree[u]:
-                continue
-            q.append(u)
-            dist[u] = time[u]
+        q = [u for u in xrange(n) if not in_degree[u]]
+        dist = [time[u] if not in_degree[u] else 0 for u in xrange(n)] 
         while q:
             new_q = []
             for u in q:
