问题描述(原题链接)
你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,其中 prerequisites[i] = [ai, bi] ,表示如果要学习课程 ai 则 必须 先学习课程 bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。
代码:
class Solution {
//拓扑排序:
//判断节点是否访问,0未访问,1访问中,2访问完成
boolean res = true;//默认结果是true
int[] state; //访问状态
ArrayList<ArrayList<Integer>> matrix; //邻接矩阵
public boolean canFinish(int numCourses, int[][] prerequisites) {
//拓扑排序
state = new int[numCourses];
matrix = new ArrayList<ArrayList<Integer>>();
for(int i=0;i<numCourses;i++){
matrix.add(new ArrayList<Integer>());
}
for(int[] temp : prerequisites){
matrix.get(temp[1]).add(temp[0]); //建立邻接关系
}
for(int i=0;i<numCourses && res;i++){
if(state[i]==0) //如果未访问
dfs(i);
}
return res;
}
private void dfs(int num){
state[num]=1; //表示访问中
for(int i : matrix.get(num)){ //访问相邻节点
if(state[i]==0){ //未访问过
dfs(i); //访问节点
if(res==false) //如果结果是false,直接返回
return;
}else if(state[i]==1){ //如果正在访问,说明有环
res=false;
return;
}
}
state[num]=2; //访问结束,则将状态置为2
}
}