时间复杂度: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图中是否有环,就判断该图拓扑序列长度是否与结点总数一致。