输入:课程数量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;}
这道题目与210的区别是210要求输入学习路线。
207代码
210代码