Lintcode 615. Course Schedule (Medium) Python

题目:https://www.lintcode.com/problem/course-schedule/description
思路:indgree = [[0,[]] for i in range(numCourses)]建立一个indegree数组,记录 [需要修的优先课程数,[是哪些科目的条件]]

class Solution:
    """
    @param: numCourses: a total of n courses
    @param: prerequisites: a list of prerequisite pairs
    @return: true if can finish all courses or false
    """
    def canFinish(self, numCourses, prerequisites):
        # write your code here

        if numCourses == 0 or prerequisites == None or len(prerequisites)==0:
            return True
        indgree = self.indgreeproduct(numCourses,prerequisites)
        queue = []
        hashtable = set()
        for i in range(numCourses):
            if indgree[i][0] == 0 :
                queue.append(i)
                hashtable.add(i)        
        while queue:
            current = queue.pop(0)
            for i in indgree[current][1]:
                indgree[i][0] -= 1
                if indgree[i][0] == 0:
                    queue.append(i)
                    hashtable.add(i)    
        if len(hashtable) == numCourses:
            return True 
        return False
            
    def indgreeproduct(self,numCourses, prerequisites):
        indgree = [[0,[]] for i in range(numCourses)]
        for i in prerequisites:
            if i[1] not in indgree[i[0]][1]:
               indgree[i[1]][0] += 1 
               indgree[i[0]][1].append(i[1])
        return indgree
    
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值