题目描述(中等难度)

207. Course Schedule - 图1

给定 n 组先修课的关系,[m,n] 代表在上 m 这门课之前必须先上 n 这门课。输出能否成功上完所有课。

解法一

把所有的关系可以看做图的边,所有的边构成了一个有向图。

对于[[1,3],[1,4],[2,4],[3,5],[3,6],[4,6]] 就可以看做下边的图,箭头指向的是需要先上的课。

207. Course Schedule - 图2

想法很简单,要想上完所有的课,一定会有一些课没有先修课,比如上图的 56。然后我们可以把 56 节点删去。

207. Course Schedule - 图3

然后 34 就可以上了,同样的道理再把 34 删去。

接下来就可以去学 12 了。因此可以完成所有的课。

代码的话,用邻接表表示图。此外,我们不需要真的去删除节点,我们可以用 outNum 变量记录所有节点的先修课门数。当删除一个节点的时候,就将相应节点的先修课个数减一即可。

最后只需要判断所有的节点的先修课门数是否全部是 0 即可。

  1. public boolean canFinish(int numCourses, int[][] prerequisites) {
  2. //保存每个节点的先修课个数,也就是出度
  3. HashMap<Integer, Integer> outNum = new HashMap<>();
  4. //保存以 key 为先修课的列表,也就是入度的节点
  5. HashMap<Integer, ArrayList<Integer>> inNodes = new HashMap<>();
  6. //保存所有节点
  7. HashSet<Integer> set = new HashSet<>();
  8. int rows = prerequisites.length;
  9. for (int i = 0; i < rows; i++) {
  10. int key = prerequisites[i][0];
  11. int value = prerequisites[i][1];
  12. set.add(key);
  13. set.add(value);
  14. if (!outNum.containsKey(key)) {
  15. outNum.put(key, 0);
  16. }
  17. if (!outNum.containsKey(value)) {
  18. outNum.put(value, 0);
  19. }
  20. //当前节点先修课个数加一
  21. int num = outNum.get(key);
  22. outNum.put(key, num + 1);
  23. if (!inNodes.containsKey(value)) {
  24. inNodes.put(value, new ArrayList<>());
  25. }
  26. //更新以 value 为先修课的列表
  27. ArrayList<Integer> list = inNodes.get(value);
  28. list.add(key);
  29. }
  30. //将当前先修课个数为 0 的课加入到队列中
  31. Queue<Integer> queue = new LinkedList<>();
  32. for (int k : set) {
  33. if (outNum.get(k) == 0) {
  34. queue.offer(k);
  35. }
  36. }
  37. while (!queue.isEmpty()) {
  38. //队列拿出来的课代表要删除的节点
  39. //要删除的节点的 list 中所有课的先修课个数减一
  40. int v = queue.poll();
  41. ArrayList<Integer> list = inNodes.getOrDefault(v, new ArrayList<>());
  42. for (int k : list) {
  43. int num = outNum.get(k);
  44. //当前课的先修课要变成 0, 加入队列
  45. if (num == 1) {
  46. queue.offer(k);
  47. }
  48. //当前课的先修课个数减一
  49. outNum.put(k, num - 1);
  50. }
  51. }
  52. //判断所有课的先修课的个数是否为 0
  53. for (int k : set) {
  54. if (outNum.get(k) != 0) {
  55. return false;
  56. }
  57. }
  58. return true;
  59. }

解法二

还有另一种思路,我们只需要一门课一门课的判断。

从某门课开始遍历,我们通过 DFS 一条路径一条路径的判断,保证过程中没有遇到环。

207. Course Schedule - 图4

深度优先遍历 1,相当于 3 条路径

1 -> 3 -> 51 -> 3 -> 61 -> 4 -> 6

深度优先遍历 2,相当于 1 条路径

2 -> 4 -> 6

深度优先遍历 3,相当于 2 条路径

3 -> 53 -> 6

深度优先遍历 4,相当于 1 条路径

4 -> 6

深度优先遍历 5,相当于 1 条路径

5

深度优先遍历 6,相当于 1 条路径

6

什么情况下不能完成所有课程呢?某条路径出现了环,如下图。

207. Course Schedule - 图5

出现了 1 -> 3 -> 6 -> 3。所以不能学完所有课程。

代码的话,用邻接表表示图。通过递归实现 DFS ,用 visited 存储当前路径上的节点。

同时用 visitedFinish 表示可以学完的课程,起到优化算法的作用。

  1. public boolean canFinish(int numCourses, int[][] prerequisites) {
  2. HashMap<Integer, ArrayList<Integer>> outNodes = new HashMap<>();
  3. HashSet<Integer> set = new HashSet<>();
  4. int rows = prerequisites.length;
  5. for (int i = 0; i < rows; i++) {
  6. int key = prerequisites[i][0];
  7. int value = prerequisites[i][1];
  8. set.add(key);
  9. if (!outNodes.containsKey(key)) {
  10. outNodes.put(key, new ArrayList<>());
  11. }
  12. //存储当前节点的所有先修课程
  13. ArrayList<Integer> list = outNodes.get(key);
  14. list.add(value);
  15. }
  16. HashSet<Integer> visitedFinish = new HashSet<>();
  17. //判断每一门课
  18. for (int k : set) {
  19. if (!dfs(k, outNodes, new HashSet<>(), visitedFinish)) {
  20. return false;
  21. }
  22. visitedFinish.add(k);
  23. }
  24. return true;
  25. }
  26. private boolean dfs(int start, HashMap<Integer, ArrayList<Integer>> outNodes, HashSet<Integer> visited,
  27. HashSet<Integer> visitedFinish) {
  28. //已经处理过或者到了叶子节点
  29. if (visitedFinish.contains(start) || !outNodes.containsKey(start)) {
  30. return true;
  31. }
  32. //出现了环
  33. if (visited.contains(start)) {
  34. return false;
  35. }
  36. //将当前节点加入路径
  37. visited.add(start);
  38. ArrayList<Integer> list = outNodes.get(start);
  39. for (int k : list) {
  40. if(!dfs(k, outNodes, visited, visitedFinish)){
  41. return false;
  42. }
  43. }
  44. visited.remove(start);
  45. return true;
  46. }

这道题本质上其实就是图的遍历。解法一是 BFS ,解法二是 DFS

解法一其实就是图的拓扑排序,解法二是判断图中是否有环的方法。

另外,图在代码中有两种表示形式,邻接表和邻接矩阵,上边的解法都采用的是邻接表。

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情