输入:课程数量n,每个课程的编号是0到n-1。课程学习的先后顺序用数组输入,例如[0,1]。要学习课程0,需要先学习课程1。
输出:如果可以学习完所有的课程,返回true,否则返回false。
分析:因为有些课程学习需要其他课程学习完才可以,这是拓扑排序的描述。我们可以根据数组对,建立一个有向图。方向是从前提课程指向能够到达的课程。需要从没有先决条件的课程i开始,接着课程i能到达的课程j入度减1。如果课程j的入度为0 ,那继续加入可学习队列。如此及不断向外扩大学习范围。可以用BFS或者DFS实现。
DFS实现:
public boolean canFinishV3(int numCourses, int[][] prerequisites) {
int[] inDegree = new int[numCourses];
boolean[] visited = new boolean[numCourses];
List<List<Integer>> graph = new ArrayList<List<Integer>>(numCourses);
for(int i=0;i<numCourses;i++){
graph.add(new ArrayList<Integer>());
}
for(int i=0;i<prerequisites.length;i++){
inDegree[prerequisites[i][0]]++;
graph.get(prerequisites[i][1]).add(prerequisites[i][0]);
}
for(int i=0;i<numCourses;i++){
if(visited[i]==false && inDegree[i]==0){
if(!bfs(graph,visited,inDegree,i)){
return false;
}
}
}
for(int i=0;i<numCourses;i++){
if(visited[i]==false) return false;
}
return true;
}
private boolean bfs(List<List<Integer>> graph,boolean[] visited,int[] inDegree,int index){
if(visited[index]) return false;
visited[index] = true;
for(Integer toCourse : graph.get(index)){
inDegree[toCourse]--;
if(inDegree[toCourse]==0){
if(!bfs(graph,visited,inDegree,toCourse)){
return false;
}
}
}
return true;
}
BFS实现
public boolean canFinishV2(int numCourses, int[][] prerequisites) {
int[] inDegree = new int[numCourses];
List<List<Integer>> graph = new ArrayList<List<Integer>>(numCourses);
for (int i = 0; i < numCourses; i++) {
graph.add(new ArrayList<Integer>());
}
for (int i = 0; i < prerequisites.length; i++) {
inDegree[prerequisites[i][0]]++;
graph.get(prerequisites[i][1]).add(prerequisites[i][0]);
}
Queue<Integer> queue = new LinkedList<Integer>();
for (int i = 0; i < numCourses; i++) {
if (inDegree[i] == 0) {
queue.add(i);
}
}
int count = 0;
while (!queue.isEmpty()) {
int course = queue.poll();
count++;
for (Integer toCourse : graph.get(course)) {
inDegree[toCourse]--;
if (inDegree[toCourse] == 0) {
queue.add(toCourse);
}
}
}
return count == numCourses;
}