题目
课程与课程之间的先决条件组成有向图,课程是图中的顶点,课程之间的先决条件是图中的有向边。对于数组 prerequisites 中的元素 [a,b],表示在学习课程 a 前必须先完成课程 b,对应从 b 指向 a 的有向边,b 是 a 的前驱课程,a 是 b 的后继课程。
如果课程之间的先决条件不存在环,则可以完成所有课程;如果课程之间的先决条件存在环,则不能完成所有课程。因此判断是否可能完成所有课程等价于判断有向图是否为无环图。
解法
判断有向图是否为无环图可以使用拓扑排序,拓扑排序可以使用广度优先搜索或深度优先搜索实现。
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
List<Integer>[] adjacentArr = new List[numCourses];
int[] indegrees = new int[numCourses];
for (int i = 0; i < numCourses; i++) {
adjacentArr[i] = new ArrayList<Integer>();
}
for (int[] prerequisite : prerequisites) {
adjacentArr[prerequisite[1]].add(prerequisite[0]);
indegrees[prerequisite[0]]++;
}
int finishCount = 0;
Queue<Integer> queue = new ArrayDeque<Integer>();
for (int i = 0; i < numCourses; i++) {
if (indegrees[i] == 0) {
queue.offer(i);
}
}
while (!queue.isEmpty()) {
int curr = queue.poll();
finishCount++;
List<Integer> adjacent = adjacentArr[curr];
for (int next : adjacent) {
indegrees[next]--;
if (indegrees[next] == 0) {
queue.offer(next);
}
}
}
return finishCount == numCourses;
}
}