0
点赞
收藏
分享

微信扫一扫

207. Course Schedule

207. Course Schedule

Description

There are a total of numCourses courses you have to take, labeled from 0 to numCourses - 1. You are given an array prerequisites where prerequisites[i] = [ai, bi] indicates that you must take course bi first if you want to take course ai.

For example, the pair [0, 1], indicates that to take course 0 you have to first take course 1.
Return true if you can finish all courses. Otherwise, return false.

Thoughts

This is a classical topological sorting question. We can use BFS to solve this kind of problems. But we have to build the graph and count the indegree first.
The graph shows the order for 2 courses – prerequisite course: unlocked courses
The indegree list shows the number of prerequsited courses for every course.
We start from the courses with indegree 0 – meaning they have no prerequisite courses. Then we do BFS from these courses. Its neighbor is the unlocked courses in the graph. For all the neighbors, their indegree can be deducted by 1 because there is one less prerequisited course for them. And now, we may some courses that have 0 indegree. We add the into the queue and do the work again until there is no course in the queue.


Codes

    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        if not prerequisites:
            return True
        '''
        build graph and indegree list first
        the given input using int to represent each course, so we can use list of list
        list_index(which is prerequisited course): [unlocking courses]
        be sure to cover all the courses
        so we use the numCourses instead of the list prerequisites to build the graph and indegree
        there maybe some courses who have no prerequisites hence not appear in prerequisites
        later we need to add/undate values into the graph/indegree
        '''
        # build adjacent courses list
        adj_course = [[] for _ in range(numCourses)]
        # number of prerequisites of each course
        preres = [0] * numCourses
        
        for unlock, prere in prerequisites:
            adj_course[prere].append(unlock)
            preres[unlock] += 1
            
        queue = collections.deque([])
   		
        for i in range(numCourses):
            if preres[i] == 0:
                queue.append(i)
        '''
        we want to know if we are able to take all the courses
        so we need to take notes on the number of courses taken
        we are not able to take all the courses in order
        if the number is less then numCourses after all the bfs done
        '''        
        taken = 0
        
        while queue:
            current = queue.popleft()  
            taken += 1
            for next_course in adj_course[current]:
                preres[next_course] -= 1
                if preres[next_course] == 0:
                    queue.append(next_course)
                    
        return taken == numCourses

Notes

This problem is asking for if we are able to take all the courses. It has other format like: return the order, if the topological order unique etc.

  • return the order
    we need to build a list to store all the courses and return it if the number of courses taken == numCourses

  • if the topological order unique
    we need to check if len(queue) == 1 in every round. len(queue) == 2 means we can learn either course in queue hence the order is not unique.

举报

相关推荐

0 条评论