现在你总共有 n 门课需要选,记为 0 到 n-1。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]
给定课程总量以及它们的先决条件,判断是否可能完成所有课程的学习?
示例 1:
输入: 2, [[1,0]] 输出: true 解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
示例 2:
输入: 2, [[1,0],[0,1]] 输出: false 解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。
思路一:入度表(广度优先遍历)
本题可简化为:课程安排图是否是 有向无环图(DAG)。即课程间规定了前置条件,但不能构成任何环路,否则课程前置条件将不成立。
通过 拓扑排序 判断此课程安排图是否是 有向无环图(DAG)。
拓扑排序是对 DAG 的顶点进行排序,使得对每一条有向边 (u, v),均有 u在排序记录中)比 v先出现。亦可理解为对某点 v而言,只有当 v的所有源点均出现了,v才能出现。
所以,思路一采用入度表,只有当一个节点入度为0,才能遍历该节点。步骤如下:
-
统计课程安排图中每个节点的入度,生成 入度表 indegrees
-
借助一个队列 queue,将所有入度为 0 的节点入队。
-
当 queue 非空时,依次将队首节点出队,在课程安排图中删除此节点 pre:
①并不是真正从邻接表中删除此节点 pre,而是将此节点对应所有邻接节点 cur 的入度 -1,即 indegrees[cur] -= 1。
②当入度 -1后邻接节点 cur 的入度为 0,说明 cur 所有的前驱节点已经被 “删除”,此时将 cur 入队。 -
在每次 pre 出队时,执行 numCourses–
-
若整个课程安排图是有向无环图(即可以安排),则所有节点一定都入队并出队过,即完成拓扑排序。换个角度说,若课程安排图中存在环,一定有节点的入度始终不为 0。
因此,拓扑排序出队次数等于课程个数,返回 numCourses == 0 判断课程是否可以成功安排。
时间复杂度:O(N+M),遍历一个图需要访问所有节点和所有临边,N 和 M分别为节点数量和临边数量;
空间复杂度:O(N),为建立入度表和队列所需额外空间。
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
//首先建立入度表
int[] indegrees = new int[numCourses];
for(int[] cp:prerequisites){
indegrees[cp[0]]++;
}
//其次将入度为0的节点入队
Queue<Integer> queue = new LinkedList<Integer>();
for(int i=0;i<numCourses;i++){
if(indegrees[i]==0) queue.offer(i);
}
//出队,并且把边从图中消除掉(邻接点入度-1),并将入度为0的邻接点入队,且numsCourse-1。
while(!queue.isEmpty()){
int index = queue.poll();
numCourses--;
for(int[] cp:prerequisites){
if(cp[1]==index){
if(--indegrees[cp[0]]==0){
queue.offer(cp[0]);
}
}
}
}
//最后判断numsCourse是否等于0
return numCourses==0?true:false;
}
}
思路二:深度优先遍历
由于此题是有向图,所以一旦在遍历过程中又再次遍历到此次DFS遍历过的节点,说明有环了。
标志位flags:若未被访问=0,被其他节点启动的DFS访问=-1,被当前节点启动的DFS访问=1。
对 numCourses 个节点依次执行 DFS,判断每个节点起步 DFS 是否存在环,若存在环直接返回 False。DFS 流程:
- 终止条件:
当 flag[i] == -1,说明当前访问节点已被其他节点启动的 DFS 访问,无需再重复搜索,直接返回 True。
当 flag[i] == 1,说明在本轮 DFS 搜索中节点 i 被第 2次访问,即 课程安排图有环,直接返回 False。 - 将当前访问节点 i 对应 flag[i] 置 1,即标记其被本轮 DFS 访问过;
- 递归访问当前节点 i 的所有邻接节点 j,当发现环直接返回 False;
- 当前节点所有邻接节点已被遍历,并没有发现环,则将当前节点 flag 置为 -1 并返回 True。
复杂度与思路一相同。
class Solution {
int[] flags;
public boolean canFinish(int numCourses, int[][] prerequisites) {
//标志位数组
flags = new int[numCourses];
boolean res = true;
for(int i=0;i<numCourses;i++){
res = dfs(i,prerequisites);
if(res == false) return res;
}
return res;
}
public boolean dfs(int index,int[][] prerequisites){
if(flags[index]==-1) return true;
if(flags[index]==1) return false;
boolean res = true;;
flags[index] = 1; //标记已经被访问过了
for(int[] cp:prerequisites){
if(cp[1]==index){
res = dfs(cp[0],prerequisites);
if(res==false) return res;
}
}
if(res==true) flags[index] = -1; //假如确定此节点不会有环,则标志位置-1.
return res;
}
}