一、需求
- 你这个学期必须选修
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 。这是不可能的。
提示:
1 <= numCourses <= 105
0 <= prerequisites.length <= 5000
prerequisites[i].length == 2
0 <= ai, bi < numCourses
prerequisites[i] 中的所有课程对 互不相同
二、BFS
2.1 思路分析
- 题目给定课程数量numCourses,课程对数组prerequisites,取第 i 个课程对:[ai,bi],它表示学习课程ai之前必须学课程bi,其中0 <= ai,bi < numCourses,现在要判断能否学完所有的课程;
- 假设编号0~8表示课程C1~C9,那么给定的课程总数为9,课程对为:[2,0]、[2,1]、[3,2]、[3,4]、[4,1]、[5,3]、[5,4]、[6,3]、[6,8]、[7,0]、[8,7];
- 对于给定的课程对,分析它们之间的联系,可以得到一张有向图(图片来源于力扣官方题解),如下图所示:
4.题目要求是判断能否学完所有的课程,也就是说根据课程对形成的有向图是否符合拓扑排序,即先修课程始终在后修课程的前面,只要图中无环,就一定是拓扑排序的;
5.所以现在问题就转变了成为判断有向图是否有环的问题,那么怎么判断呢?
6.我们可以利用队列来存储入度为0的节点,表示该节点为先修课程,每次出队该节点,该节点指向的邻接点的入度要减1,当邻接点的入度为0时,加入队列,当队列为空时,检查出队的节点数量是否与课程数相等即可;
7.比较关键的地方是如何统计节点入度,如何统计节点指向的所有邻接点,在下面的代码中给出;
2.2 代码实现
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
if(prerequisites.length == 0) {
return true;
}
//定义入度数组
int[] inDegree = new int[numCourses];
//定义邻接点集合数组,adj中有numCourses个HashSet对象
HashSet<Integer>[] adj = new HashSet[numCourses];
//初始化集合中的每个HashSet对象
for(int i = 0; i < numCourses; i++) {
adj[i] = new HashSet<>();
}
//获取课程对
for(int[] p : prerequisites) {
//统计课程的入度
inDegree[p[0]]++;
//存储课程相应的邻接点
adj[p[1]].add(p[0]);
}
Queue<Integer> queue = new LinkedList<>();
//将入度为0的课程加入队列
for(int i = 0; i < numCourses; i++) {
if(inDegree[i] == 0) {
queue.add(i);
}
}
//记录出队的课程数
int count = 0;
while(queue.size() != 0) {
int course = queue.poll();
count++;
for(int adjCourse : adj[course]) {
inDegree[adjCourse]--;
if(inDegree[adjCourse] == 0) {
queue.add(adjCourse);
}
}
}
return count == numCourses;
}
}
2.3 复杂度分析
- 时间复杂度为O(m+n),m为课程对数,n为节点数;
- 空间复杂度为O(m+n);
三、学习地址
作者:liweiwei1419
链接:https://leetcode-cn.com/problems/course-schedule/solution/tuo-bu-pai-xu-by-liweiwei1419/