There are a total of n courses you have to take, labeled from 0
to n - 1
.
Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1]
Given the total number of courses and a list of prerequisite pairs, is it possible for you to finish all courses?
Example
Given n = 2
, prerequisites = [[1,0]]
Return true
Given n = 2
, prerequisites = [[1,0],[0,1]]
return false;
java
public class Solution {
/*
* @param numCourses: a total of n courses
* @param prerequisites: a list of prerequisite pairs
* @return: true if can finish all courses or false
*/
public boolean canFinish(int numCourses, int[][] prerequisites) {
// write your code here
if (numCourses <= 0 || prerequisites == null) {
return true;
}
int[] degree = new int[numCourses];
List[] edges = new ArrayList[numCourses];
for (int i = 0; i < numCourses; i++) {
edges[i] = new ArrayList<Integer>();
}
for (int i = 0; i < prerequisites.length; i++) {
int m = prerequisites[i][0];
int n = prerequisites[i][1];
degree[m]++;
edges[n].add(m);
}
int count = 0;
Queue<Integer> queue = new LinkedList<>();
for (int i = 0; i < degree.length; i++) {
if (degree[i] == 0) {
queue.offer(i);
count++;
}
}
while (!queue.isEmpty()) {
int node = queue.poll();
for (Object val : edges[node]) {
degree[(Integer)val]--;
if (degree[(Integer)val] == 0) {
queue.offer((Integer)val);
count++;
}
}
}
return count == numCourses;
}
}