207. Course Schedule 难度:medium 类别:图

题目:

There are a total of n courses you have to take, labeled from 0 to n - 1.

Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1]

Given the total number of courses and a list of prerequisite pairs, is it possible for you to finish all courses?

For example:

2, [[1,0]]
There are a total of 2 courses to take. To take course 1 you should have finished course 0. So it is possible.

2, [[1,0],[0,1]]
There are a total of 2 courses to take. To take course 1 you should have finished course 0, and to take course 0 you should also have finished course 1. So it is impossible.


思路:

此题要我们判断给定的图能否进行拓扑排序,就用拓扑排序的一般方法对图进行排序,看最终序列的元素个数是否等于节点数,如果相等则可以进行拓扑排序,否则则不能。


程序:

def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        queue = []
        result = []

        adj = [[] for i in range(numCourses)]
        indegree = [0 for i in range(numCourses)]

        for l in prerequisites:
            adj[l[1]].append(l[0])
            indegree[l[0]] += 1

    
        for i in range(numCourses):
            if indegree[i] == 0:
                queue.append(i)

        while len(queue) != 0:
            u = queue.pop(0)
            result.append(u)
        
            for v in adj[u]:
                indegree[v] -= 1
                if indegree[v] == 0:
                    queue.append(v)

        if len(result) == numCourses:
            return True

        else:
            return False


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值