你这个学期必须选修 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;
}