Course Schedule
DESCRIPTION (credit Leetcode.com)
You have to take a total of numCourses courses, which are labeled from 0 to numCourses - 1. You are given a list of prerequisites pairs, where prerequisites[i] = [a, b] indicates that you must complete course b before course a.
Given the total number of courses and a list of prerequisite pairs, write a function to determine if it is possible to finish all courses.
EXAMPLES
Example 1:
Input:
numCourses = 3 prerequisites = [[1, 0], [2, 1]]
Output: true
Explanation: You can take courses in the following order: 0, 1, 2.
Example 2:
Input:
numCourses = 3 prerequisites = [[1, 0], [0, 1],[1,2]]
Output: false
Explanation: It is impossible to finish all courses, as you must finish course 1 before course 0 and course 0 before course 1.
Run your code to see results here
Explanation
We can use the following approach to solve this problem:
- Start with a course that has no prerequisites.
- If that course is a prerequisite for another course, remove it from the prerequisites list.
- Now, Find the next course that has no prerequisites, and repeat the process.
When we can't find any more courses with no prerequisites, there are two possibilities:
- If we have already taken all the courses, then we can return true.
- Otherwise, we return false (as this means there was a circular dependency in the prerequisites).
Here's a visual representation of the above approach for the input numCourses = 4 and prerequisites = [[1,0],[2,1],[3,1],[2,3]]:
We've represented the input as a graph, where each node represents a course, and the edges represent the prerequisites between the courses.
At this point, there are no more courses without prerequisites, and we've visited all the courses. So, we can return true.
Cycle
Let's instead consider the input numCourses = 5 and prerequisites = [[1,0],[2,1],[3,2],[4,3],[1,4]]:
We start with Node 0, like before:
But now, when we remove Node 0, we can't find any more courses without prerequisites!
At this point, we've only processed 1 course, but there are still 4 courses left. So, we return false.
Topological Sort
The above approach is an application of Topological Sort. If we start by calculating the in-degree of each node, we can find the nodes with an in-degree of 0 and add them to a queue.
Next, while the queue is not empty, we remove a node from the queue. After removing a node, we decrement the in-degree of its neighbors. If any of the neighbors have an in-degree of 0, we add them to the queue.
Each time we remove a node from the queue, we increment a counter. When the queue is finally empty, we check if the counter is equal to the number of courses. If it is, we return true. Otherwise, we return false.
from collections import defaultdict, dequeclass Solution:def canFinish(self, numCourses, prerequisites):graph = defaultdict(list)in_degree = [0] * numCoursesfor dest, src in prerequisites:graph[src].append(dest)in_degree[dest] += 1queue = deque([i for i in range(numCourses) if in_degree[i] == 0])courses_taken = 0while queue:course = queue.popleft()courses_taken += 1for neighbor in graph[course]:in_degree[neighbor] -= 1if in_degree[neighbor] == 0:queue.append(neighbor)return courses_taken == numCourses
Complexity Analysis
The time complexity for this approach is O(V + E), where V is the number of courses and E is the number of prerequisites. This is because we visit each node and edge once.
The space complexity is O(V + E) as well, since we store the graph as an adjacency list and the in-degree of each node.
Loading comments...