你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们:[0,1]
给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?
示例 1:
输入: 2, [[1,0]]
输出: true
解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
DFS代码
class Solution {
boolean[] finished;public boolean canFinish(int numCourses, int[][] prerequisites) {finished=new boolean[numCourses];HashMap<Integer,List<Integer>> map=new HashMap<>();//记录课程学完后可以学习的课程HashMap<Integer,HashSet<Integer>> map2=new HashMap<>();//记录需要先修的课程for(int i=0;i<numCourses;i++)//创建邻接表{map.put(i,new ArrayList<>());map2.put(i,new HashSet<>());}for(int[] pre:prerequisites){map.get(pre[1]).add(pre[0]);map2.get(pre[0]).add(pre[1]);}for(int i=0;i<numCourses;i++)Finish(i,map,map2);for(int i=0;i<numCourses;i++)//检查是不是全部学完了if(!finished[i]) return false;return true;}public void Finish(int cur,HashMap<Integer,List<Integer>> map, HashMap<Integer,HashSet<Integer>> map2) {//深度优先搜索if(!map2.get(cur).isEmpty()) return;finished[cur]=true;for(int next:map.get(cur)){map2.get(next).remove(cur);Finish(next,map,map2);}}
}
BFS代码
public boolean canFinish(int numCourses, int[][] prerequisites) {Queue<Integer> queue=new LinkedList<>();int[] adj=new int[numCourses];Map<Integer,Set<Integer>> pre=new HashMap<>();for(int i=0;i<numCourses;i++)pre.put(i,new HashSet<>());for(int[] t:prerequisites){pre.get(t[1]).add(t[0]);//邻接表adj[t[0]]++;//记录节点的入度}for(int i=0;i<numCourses;i++)//入度为0的节点入队if(adj[i]==0) queue.add(i);while (!queue.isEmpty()){int cur=queue.poll();for(int c:pre.get(cur))//相邻节点入度减1{if(--adj[c]==0) queue.offer(c);//入度为0的节点入队}}for(int i=0;i<numCourses;i++)if(adj[i]>0) return false;//还存在不能消除的边return true;}