Leetcode 207 题 课程表

题目描述

你这个学期必须选修 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。这是不可能的。

思路分析

「简单描述下题目要求」

① 给出 n 门课程，分别记为 0 ~ n-1。
② 给出先决条件，例如 [0,1]，表示想学课程 0，需要先学课程 1。
③ 判断能否完成所有课程。

「何时不能完成所有课程？」

根据题目描述，我们可以很容易将例 1 和例 2 以有向图的形式表现出来。

可以看到，例 2 所绘制的图具有环，即想学课程 1，需要先学课程 0，而想学课程 0，却得先学课程 1，所以不能完成所有课程。

「举个例子」

为了方便分析，这里举个例子：共有 4 门课程，其先决条件为[1,0]、[2,0]、[3,1]、[3,2]。

「如何完成每门课程？」

首先需要先学习课程 0，因为 0 没有先决条件；其次再学习 1 和 2 中的任意课程，并没有顺序上的要求；只有当课程 1 和 2 全部学完，才能再学习课程 3。

为了完成以上步骤，我们需要存储每个节点的入度和绘制邻接表。

入度： 简单的说就是当前节点被多少个其他节点所指向。

节点 0 的入度为 0。节点1 和节点 2 的入度为 1。节点 3 的入度为 2。

邻接表： 保存图中每个节点所指向节点的信息。

于是，以上过程将变为：

1、查找入度为 0 的节点，移除该节点，查找该节点的邻接表，将该节点所指向节点的入度减 1（因为该节点被移除）。

2、直到再也找不到入度为 0 的节点。

3、所移除节点的个数（学习的课程数）和题目所给出的课程数一致，说明可以学完所有课程。如果不一致，说明存在环，导致课程不能全部学完。

代码描述

使用 Java 进行代码描述：

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        int[] inDegree = new int[numCourses]; // 存储入度
        List<List<Integer>> edges = new ArrayList<>(); // 构建邻接表

        // 初始化入度和邻接表
        for (int i = 0; i < numCourses; i++) {
            edges.add(new ArrayList<>());
        }

        for (int[] pre : prerequisites) {
            inDegree[pre[0]]++; // 记录入度
            edges.get(pre[1]).add(pre[0]);
        }

        Queue<Integer> queue = new LinkedList();
        // 将没有前置课程的课程编号放入队列中
        for (int i = 0; i < numCourses; i++) {
            if (inDegree[i] == 0) {
                queue.add(i);
            }
        }

        // 记录能完成的选修课程的数量
        int count = 0;
        while (!queue.isEmpty()) {
            count++;
            List<Integer> list = edges.get(queue.poll());
            if (list == null) continue;

            for (int val : list) {
                int num = --inDegree[val];
                // 入度为0，继续加入到队列中
                if (num == 0) queue.offer(val);
            }
        }

        return count == numCourses;
    }
}

技术公众号：小猿君的算法笔记