题目描述
你这个学期必须选修 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。这是不可能的。
思路分析
「简单描述下题目要求」
① 给出 n 门课程,分别记为 0 ~ n-1。
② 给出先决条件,例如 [0,1],表示想学课程 0,需要先学课程 1。
③ 判断能否完成所有课程。
「何时不能完成所有课程?」
根据题目描述,我们可以很容易将例 1 和例 2 以有向图的形式表现出来。
可以看到,例 2 所绘制的图具有环,即想学课程 1,需要先学课程 0,而想学课程 0,却得先学课程 1,所以不能完成所有课程。
「举个例子」
为了方便分析,这里举个例子:共有 4 门课程,其先决条件为[1,0]、[2,0]、[3,1]、[3,2]。
「如何完成每门课程?」
首先需要先学习课程 0,因为 0 没有先决条件;其次再学习 1 和 2 中的任意课程,并没有顺序上的要求;只有当课程 1 和 2 全部学完,才能再学习课程 3。
为了完成以上步骤,我们需要存储每个节点的入度和绘制邻接表。
入度: 简单的说就是当前节点被多少个其他节点所指向。
节点 0 的入度为 0。节点1 和节点 2 的入度为 1。节点 3 的入度为 2。
邻接表: 保存图中每个节点所指向节点的信息。
于是,以上过程将变为:
1、查找入度为 0 的节点,移除该节点,查找该节点的邻接表,将该节点所指向节点的入度减 1(因为该节点被移除)。
2、直到再也找不到入度为 0 的节点。
3、所移除节点的个数(学习的课程数)和题目所给出的课程数一致,说明可以学完所有课程。如果不一致,说明存在环,导致课程不能全部学完。
代码描述
使用 Java 进行代码描述:
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
int[] inDegree = new int[numCourses]; // 存储入度
List<List<Integer>> edges = new ArrayList<>(); // 构建邻接表
// 初始化入度和邻接表
for (int i = 0; i < numCourses; i++) {
edges.add(new ArrayList<>());
}
for (int[] pre : prerequisites) {
inDegree[pre[0]]++; // 记录入度
edges.get(pre[1]).add(pre[0]);
}
Queue<Integer> queue = new LinkedList();
// 将没有前置课程的课程编号放入队列中
for (int i = 0; i < numCourses; i++) {
if (inDegree[i] == 0) {
queue.add(i);
}
}
// 记录能完成的选修课程的数量
int count = 0;
while (!queue.isEmpty()) {
count++;
List<Integer> list = edges.get(queue.poll());
if (list == null) continue;
for (int val : list) {
int num = --inDegree[val];
// 入度为0,继续加入到队列中
if (num == 0) queue.offer(val);
}
}
return count == numCourses;
}
}