题目
你这个学期必须选修 numCourses
门课程,记为 0
到 numCourses - 1
。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites
给出,其中 prerequisites[i] = [ai, bi]
,表示如果要学习课程 ai
则 必须 先学习课程 bi
。
-
例如,先修课程对
[0, 1]
表示:想要学习课程0
,你需要先完成课程1
。
请你判断是否可能完成所有课程的学习?如果可以,返回 true
;否则,返回 false
。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]] 输出:true 解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。
示例 2:
输入:numCourses = 2, prerequisites = [[1,0],[0,1]] 输出:false 解释:总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。
思路
非常经典的拓扑排序题。
首先明确两点:
-
如果图G存在环,那么G就不存在拓扑排序;
-
如果图G是有向无环图,那么它的拓扑排序可能不止一种;
采用广度优先遍历
代码如下:
class Solution(object):
def canFinish(self, numCourses, prerequisites):
inDegree = [0 for _ in range(numCourses)]
Adj = [[] for _ in range(numCourses)]
for i,j in prerequisites:
inDegree[j] += 1
Adj[i].append(j)
q = deque()
n = numCourses
for i in range(numCourses):
if inDegree[i] == 0:
q.append(i)
n -= 1
while q:
cur = q.popleft()
for i in Adj[cur]:
inDegree[i] -= 1
if inDegree[i] == 0:
q.append(i)
n -= 1
return n == 0
时间复杂度:O(n+m)
空间复杂度:O(n+m)
另外还有一道题课程表II的题,210,需要给出遍历顺序,那么就在队列拿出来的时候维护一个答案数组即可
代码如下:
class Solution(object):
def findOrder(self, numCourses, prerequisites):
#跟课程表其实很像,维护一个答案数组即可…………,广度用队列,要不就dfs用递归
lingjie = [[] for _ in range(numCourses)]
rudu = [0]*numCourses
zero = deque()
ans = []
num = numCourses
for i,j in prerequisites:
lingjie[j].append(i)
rudu[i] += 1
for i in range(numCourses):
if rudu[i] == 0:
zero.append(i)
num -= 1
while zero:
cur = zero.popleft()
ans.append(cur)
for i in lingjie[cur]:
rudu[i] -= 1
if rudu[i] == 0:
zero.append(i)
num -= 1
return ans if num ==0 else []
时空复杂度一致