Update 1462. Course Schedule IV.py

Author: amineshx

1462. Course Schedule IV/1462. Course Schedule IV.py

@@ -23,4 +23,9 @@ def checkIfPrerequisite(self, numCourses: int, prerequisites: List[List[int]], q
                 if indegree[next_course] == 0:
                     q.append(next_course)
 
-        return [q[0] in mp[q[1]] for q in queries]
+        return [q[0] in mp[q[1]] for q in queries]
+
+sol = Solution()
+print(sol.checkIfPrerequisite(numCourses = 2, prerequisites = [[1,0]], queries = [[0,1],[1,0]]))
+print(sol.checkIfPrerequisite(numCourses = 2, prerequisites = [], queries = [[1,0],[0,1]]))
+print(sol.checkIfPrerequisite(numCourses = 3, prerequisites = [[1,2],[1,0],[2,0]], queries = [[1,0],[1,2]]))