207. 课程表
你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们:[0,1]
给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?
示例 1:
输入: 2, [[1,0]] 输出: true 解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。 示例
2:输入: 2, [[1,0],[0,1]] 输出: false 解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成课程
0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。提示:
输入的先决条件是由 边缘列表 表示的图形,而不是 邻接矩阵 。详情请参见图的表示法。 你可以假定输入的先决条件中没有重复的边。 1 <=
numCourses <= 10^5 通过次数35,982提交次数71,139 在真实的面试中遇到过这道题?
拓扑排序
class Solution:
def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
dic={}
temp=[0]*numCourses
count=0
for prerequisite in prerequisites:
if prerequisite[1]not in dic:
dic[prerequisite[1]]=[prerequisite[0]]
else:
dic[prerequisite[1]].append(prerequisite[0])
temp[prerequisite[0]]+=1
temp2=[ i for i in range(len(temp))if temp[i]==0]
while temp2:
node=temp2.pop(0)
count+=1
if node not in dic:
continue
for j in dic[node]:
temp[j]-=1
if temp[j]==0:
temp2.append(j)
return count==numCourses
- 就是按照老师上课讲的,首先我使用了一个字典的结构来储存图
- 然后一个列表temp2储存不需要先修课程的课程,即入度为零的课程
- 然后一个列表temp来储存所有的课程所需要的先修课程的数量
- 最后我们只需要看从temp2里面弄出来的节点是不是等于numCourses就知道该学生能否完成所有的课程了,即图中有无闭环。
深度优先搜索
class Solution:
def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
dic={}
visited=set()
for prerequisite in prerequisites:
if prerequisite[1]not in dic:
dic[prerequisite[1]]=[prerequisite[0]]
else:
dic[prerequisite[1]].append(prerequisite[0])
def dfs(i,visiting):
if i in visiting:
return False
if i in visited:
return True
visited.add(i)
visiting.add(i)
if i not in dic:
visiting.remove(i)
return True
for j in dic[i]:
if not dfs(j,visiting):
return False
visiting.remove(i)
return True
for i in range(numCourses):
if i in visited:
continue
if not dfs(i,set()):
return False
return True
- 该学生能否完成所有的课程其实就是看这个图中有没有环就可以了
- 这样我们在这里使用深度优先搜索,查看是否有环
1.具体过程就是,我们在遍历节点的时候遍历完了就标记到visited里面,遍历中就标记到visiting里面,假如我们遍历到一个被标记到visiting的节点
说明我们又绕回来了,说明有环,return False。