feat(work): implement Course Schedule II solution

This commit is contained in:
Prad Nukala committed 2026-09-22 15:02:31 -04:00
1 parent ca73d3235c
commit 307562994b
2 files changed
+70 -6

No files matched your search

@@ -36,4 +36,36 @@ sidebar:
```py ```py
class Solution: class Solution:
def findOrder(self, numCourses: int, prerequisites: list[list[int]]) -> list[int]: def findOrder(self, numCourses: int, prerequisites: list[list[int]]) -> list[int]:
prereq = {c: [] for c in range(numCourses)}
for crs, pre in prerequisites:
prereq[crs].append(pre)
# course has 3 possible states:
# visited -> crs has been added to output
# visiting -> crs not added to output, but added to cycle
# unvisited -> crs not added to output nor cycle
output = []
visit, cycle = set(), set()
def dfs(crs):
if crs in cycle:
return False
if crs in visit:
return True
cycle.add(crs)
for pre in prereq[crs]:
if dfs(pre) == False:
return False
cycle.remove(crs)
visit.add(crs)
output.append(crs)
return True
for c in range(numCourses):
if dfs(c) == False:
return []
return output
``` ```
@@ -46,19 +46,51 @@ Output: [0]
Constraints: Constraints:
• 1 <= numCourses <= 2000 • 1 <= numCourses <= 2000
• 0 <= prerequisites.length <= numCourses * (numCourses - 1) • 0 <= prerequisites.length <= numCourses * (numCourses - 1)
• prerequisites[i].length == 2 • prerequisites[i].length == 2
• 0 <= ai, bi < numCourses • 0 <= ai, bi < numCourses
• ai != bi • ai != bi
• All the pairs [ai, bi] are distinct. • All the pairs [ai, bi] are distinct.
""" """
class Solution: class Solution:
def findOrder(self, numCourses: int, prerequisites: list[list[int]]) -> list[int]: def findOrder(self, numCourses: int, prerequisites: list[list[int]]) -> list[int]:
prereq = {c: [] for c in range(numCourses)}
for crs, pre in prerequisites:
prereq[crs].append(pre)
# course has 3 possible states:
# visited -> crs has been added to output
# visiting -> crs not added to output, but added to cycle
# unvisited -> crs not added to output nor cycle
output = []
visit, cycle = set(), set()
def dfs(crs):
if crs in cycle:
return False
if crs in visit:
return True
cycle.add(crs)
for pre in prereq[crs]:
if dfs(pre) == False:
return False
cycle.remove(crs)
visit.add(crs)
output.append(crs)
return True
for c in range(numCourses):
if dfs(c) == False:
return []
return output