LeetCode0207.课程表 Go语言AC笔记

时间复杂度：O(n)

解题思路

先根据课程关系建立图，然后对该图找一个拓扑序列，检查拓扑序列的长度是否与结点总数一致，若不一致则说明图中有环，故无法完成所有课程的学习。

建立图需要一个edges二维数组和一个in一维数组分别保存每个结点的所有相邻结点和每个结点的入度值。

在寻找一个拓扑序列的过程中，需要一个队列来维护所有入度值为0的结点，每次都访问队头结点，然后将其相邻结点入度值减1，如果入度值减为0了那就入队等待访问。

最后判断拓扑序列的长度是否与结点总数一致，一致返回true，不一致返回false。

AC代码

func canFinish(numCourses int, prerequisites [][]int) bool {
    edges,in,res:=make([][]int,numCourses),make([]int,numCourses),[]int{}//结点的相邻结点、结点的入度、一个拓扑序列
    //建立图
    for _,v:=range prerequisites{
        edges[v[1]]=append(edges[v[1]],v[0])
        in[v[0]]++
    }
    q:=[]int{}//存放待访问结点的队列
    //初始化队列，入度为0的结点入队
    for i,v:=range in{
        if v==0{
            q=append(q,i)
        }
    }
    //只要队列中有结点就找相邻结点
    for len(q)>0{
        front:=q[0]
        q=q[1:]
        res=append(res,front)//队头结点已经遍历，将其加入到拓扑序列中
        //遍历队头的所有相邻结点
        for _,v:=range edges[front]{
            in[v]--//入度减1
            if in[v]==0{
                q=append(q,v)//入度为0就入队
            }
        }
    }
    return len(res)==numCourses
}

感悟

很经典的一道考察拓扑序列的题目。如果判断AOE图中是否有环，就判断该图拓扑序列长度是否与结点总数一致。