课程表
· 题目描述:
你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-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。这是不可能的。
提示:
输入的先决条件是由 边缘列表 表示的图形,而不是 邻接矩阵 。详情请参见图的表示法。
你可以假定输入的先决条件中没有重复的边。
1 <= numCourses <= 10^5
· 思路:
概括来说就是用深度优先搜索(DFS) 判断图中是否有环。
将每一门课程作为一个节点,首先定义一个二维列表,第 i 行存放课程 i 的子女节点。
通过一个 flags列表来判断每个节点的被访问状态:
- 0表示该节点未被任何节点启动的 DFS 访问;
- -1表示该节点已被其他节点启动的 DFS 访问;
- 1表示该节点已被当前节点启动的 DFS 访问。
对 numCourses 个节点(i)依次执行 DFS,判断每个节点起步的 DFS 是否存在环,若存在环直接返回 false。
若flags[i] == -1,说明当前访问节点已被其他节点启动的 DFS 访问,直接返回true即可。
若flags[i] == 1,说明在本轮 DFS 搜索中节点 i 已被访问过,即图中出现了环 ,直接返回 false。
将当前访问节点 i 对应 flags[i] 置 1,即标记其被本轮 DFS 访问过;
递归访问当前节点 i 的所有邻接节点 j,当发现环直接返回false。
若当前节点所有邻接节点已被遍历,并没有发现环,则将当前节点 flags[i] 置为 −1 并返回 true。
若整个图 DFS 结束并未发现环,返回 true。
· 代码:
class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
List<List<Integer>>lst=new ArrayList<>();
for(int i=0;i<numCourses;i++)
{
lst.add(new ArrayList<>());
}
int[]flags=new int[numCourses];
for(int[]kk : prerequisites)
{
lst.get(kk[1]).add(kk[0]);
}
for(int i=0;i<numCourses;i++)
{
if(!dfs(lst,flags,i))return false;
}
return true;
}
private boolean dfs(List<List<Integer>>lst,int[]flags,int i)
{
if(flags[i]==1)return false;
if(flags[i]==-1)return true;
flags[i]=1;
for(Integer j : lst.get(i))
{
if(!dfs(lst,flags,j))return false;
}
flags[i]=-1;
return true;
}
}