与跟着专注于计算机视觉的AndyJ的妈妈我学算法之每日一题leetcode1162地图分析
相似,都是BFS。
BFS还是挺重要的,需要多加看,多加写。
题目:
你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们:[0,1]
给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/course-schedule
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
直接BFS。注意的是,入度list的构建; adjacent的构建。心里要知道这样构建的原因。也就是这个结构,能带来什么便利。
好了,上代码
对了,要多回头看,多回头写。
class Solution:
def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
indegree = [0 for i in range(numCourses)]
adjacent = [[] for i in range(numCourses)]
for cur, pre in prerequisites:
adjacent[pre].append(cur)
indegree[cur] += 1
queue = []
for i in range(numCourses):
if indegree[i]==0:
queue.append(i)
while queue:
ind = queue.pop(0)
numCourses -= 1
for node in adjacent[ind]:
indegree[node] -= 1
if indegree[node] == 0:
queue.append(node)
return numCourses==0
这个题,心中一定要有图的样子。来构建indegree和adjacent。
Tags : depth-first-search | breadth-first-search | graph | topological-sort
好了,就这样了。