你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,其中 prerequisites[i] = [ai,bi] ,表示如果要学习课程 ai 则 必须 先学习课程 bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。 请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。
先贴一下错误代码:(思路,使用map存储有入度的节点,将入度为0的点入队,出队的点为前置节点,遍历数组判断加入后置节点)
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
Map<Integer,Integer> mp = new HashMap<>();
Queue<Integer> queue = new LinkedList<>();
for(int i = 0; i < prerequisites.length; i++){
mp.put(prerequisites[i][1],1);
}
for(int i = 0; i < numCourses; i++){
if(!mp.containsKey(i)){
queue.offer(i);
}
}
while(!queue.isEmpty()){
int x = queue.peek();
queue.poll();
if(mp.containsKey(x)){
mp.remove(x);
}
for(int i = 0; i < prerequisites.length; i++){
if(prerequisites[i][0] == x && !mp.containsKey(x)){
queue.offer(prerequisites[i][1]);
}
}
}
if(mp.isEmpty())
return true;
return false;
}
}
上述代码的问题,无法处理有环出现的情况。
正确代码如下:按a->b记录边,同时记录每个节点的入度,判断是否可行时要求此节点的入度为0,同时以此节点为头的边界点入度减一,满足入度为0继续入队
class Solution {
List<List<Integer>> edges;
int[] indeg;
public boolean canFinish(int numCourses, int[][] prerequisites) {
edges = new ArrayList<List<Integer>>();
for (int i = 0; i < numCourses; ++i) {
edges.add(new ArrayList<Integer>());
}
indeg = new int[numCourses];
for (int[] info : prerequisites) {
edges.get(info[1]).add(info[0]);
++indeg[info[0]];
}
Queue<Integer> queue = new LinkedList<Integer>();
for (int i = 0; i < numCourses; ++i) {
if (indeg[i] == 0) {
queue.offer(i);
}
}
int visited = 0;
while (!queue.isEmpty()) {
++visited;
int u = queue.poll();
for (int v: edges.get(u)) {
--indeg[v];
if (indeg[v] == 0) {
queue.offer(v);
}
}
}
return visited == numCourses;
}
}