207. 课程表

huanxiangcoco2024-09-22 16:37

思路

想象成有向图一个结点代表一门课程

d表示所有课程的集合

dict1用来装有入度的结点以及对应的入度数（如：[0,1] 学了1才能学0：即1指向0，即只有0有入度所以prerequisites每个值i的i[0]即为有入度的结点）

q表示当前层无入度的结点（入度为0）

nex：表示下一层无入度的结点

第一层：当前prerequisites无入度的结点

在遍历无入度结点时（代表选这门课程），因为有指向，所以对应指向值的入度-1，当对应指向值入度为0时即作为下一层的无入度结点循环遍历

遍历结束后，若dict1中有入度不为0的，说明这门课无法被选

python 复制代码

class Solution(object):
    def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        if len(prerequisites)==0 or len(prerequisites)==1:
            return True
        #课程集合
        d=set()
        #有入度的集合
        dict1={}
        for i in prerequisites:
            if i[0] in dict1:
                dict1[i[0]]+=1
            else:
                dict1[i[0]]=1
            d.add(i[0])
            d.add(i[1])
        
        q =[]
        #第一层要遍历的数（根结点），入度为0
        for i in d:
            if i not in dict1:
                q.append(i)
        #无入度为0的结点，成环 [[1,0],[0,1]]
        if q==[]:
            return False
        while q:
            nex=[]
            for i in q:
                for j in prerequisites:
                    if i==j[1]:
                        dict1[j[0]]-=1
                        if dict1[j[0]]==0:
                            nex.append(j[0])
            q=nex
        for i in dict1:
            if dict1[i]!=0:
                return False
        return True