力扣题目
解题思路
java代码
力扣题目:
你这个学期必须选修 numCourses
门课程,记为 0
到 numCourses - 1
。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites
给出,其中 prerequisites[i] = [ai, bi]
,表示如果要学习课程 ai
则 必须 先学习课程 bi
。
- 例如,先修课程对
[0, 1]
表示:想要学习课程0
,你需要先完成课程1
。
请你判断是否可能完成所有课程的学习?如果可以,返回 true
;否则,返回 false
。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]] 输出:true 解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。
示例 2:
输入:numCourses = 2, prerequisites = [[1,0],[0,1]] 输出:false 解释:总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。
解题思路:
算法原理:
这道题使用深度优先搜索(DFS)来判断给定的课程安排是否存在环,如果存在环则无法完成所有课程,否则可以完成。
思路:
- 首先创建一个邻接表
edges
来存储课程之间的依赖关系。 - 创建一个标记数组
visited
来记录每个课程的访问状态(0 表示未访问,1 表示正在访问,2 表示已访问完成)。 - 然后对每个未访问的课程进行 DFS 搜索。
- 在 DFS 过程中,如果遇到正在访问的节点,说明存在环,将
valid
标记为false
并返回;如果遇到已访问完成的节点则跳过;如果是未访问的节点则继续递归搜索。 - 当 DFS 完成一个节点的子树搜索后,将该节点标记为已访问完成。
代码分析:
canFinish
方法中,初始化邻接表和标记数组。然后遍历每个课程,如果未访问则进行 DFS 搜索。dfs
方法中,首先将当前节点标记为正在访问。然后遍历其依赖的课程,如果是未访问则递归 DFS,如果是正在访问则说明有环,设置valid
为false
并返回,如果是已访问完成则跳过。最后将当前节点标记为已访问完成。
时间复杂度:O(m + n),其中是先决条件的数量,是课程数量。
空间复杂度:O(m + n),主要用于存储邻接表和标记数组。
java代码:
package org.example.mouth8;
import java.util.ArrayList;
import java.util.List;
public class Leetcode207 {
public static void main(String[] args) {
int[][] prerequisites = {{1, 0}};
boolean b = new Leetcode207().canFinish(2, prerequisites);
System.out.println(b);
}
List<List<Integer>> edges;
int[] visited;
boolean valid = true;
public boolean canFinish(int numCourses, int[][] prerequisites) {
edges = new ArrayList<List<Integer>>();
for (int i = 0; i < numCourses; ++i) {
edges.add(new ArrayList<Integer>());
}
visited = new int[numCourses];
for (int[] info : prerequisites) {
edges.get(info[1]).add(info[0]);
}
for (int i = 0; i < numCourses && valid; ++i) {
if (visited[i] == 0) {
dfs(i);
}
}
return valid;
}
public void dfs(int u) {
visited[u] = 1;
for (int v: edges.get(u)) {
if (visited[v] == 0) {
dfs(v);
if (!valid) {
return;
}
} else if (visited[v] == 1) {
valid = false;
return;
}
}
visited[u] = 2;
}
}
更多详细内容同步到公众号,感谢大家的支持!
每天都会给刷算法的小伙伴推送明日一题,并且没有任何收费项