题目:
你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,其中 prerequisites[i] = [ai, bi] ,表示如果要学习课程 ai 则 必须 先学习课程 bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。
题解:
几个重要变量:
1.入度数组:存放每个课程的入度数,0位置即为课程0的入度数
2.邻接表:用list就行,不用map,存放每个课程的邻接关系,也就是依靠课程0、1...的其他课程的list组成的list
3.队列:存放入度为0的课程
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
int[] indexgrees = new int[numCourses]; //入度表,对应位置就是课程序号
List<List<Integer>> adj = new ArrayList<>(); //邻接表,对应位置就是课程序号,比如0位置是课程0的邻接关系的list
Queue<Integer> queue = new LinkedList<>(); //存放入度为0的课程,ArrayList不能实现queue
//把所有课程的邻接关系list初始化
for (int i = 0; i < numCourses; i++) {
adj.add(new ArrayList<>());
}
//更新入度表和邻接表
for (int[] cp : prerequisites) { //例:cp为[3,0],先有0才有3
indexgrees[cp[0]]++; //入度表的3位置增加入度
adj.get(cp[1]).add(cp[0]); //adj.get(cp[1])是课程0的邻接关系list,将依靠0的课程3加入
}
//遍历入度表,入度为0的加到队列queue中
for (int i = 0; i < numCourses; i++) {
if (indexgrees[i] == 0) {
queue.offer(i);
}
}
while (!queue.isEmpty()) {
int pre = queue.poll();
numCourses--;
//遍历被移除队列的课程的邻接关系list
for (int cur : adj.get(pre)) {
indexgrees[cur]--; //list中的课程入度减一
if (indexgrees[cur] == 0) { //如果入度变成0,加入queue中
queue.offer(cur);
}
}
}
return numCourses == 0; //出队次数等于课程个数,课程数为0,则无环
}
}
参考: