LeetCode 207. 课程表

题目描述

这个学期需要选修 numCourses 门课程,课程编号为 0numCourses - 1

数组 prerequisites 表示课程之间的先修关系,其中 prerequisites[i] = [ai, bi] 表示:如果要学习课程 ai,必须先学习课程 bi

例如:

复制代码
[0, 1]

表示想学习课程 0,需要先完成课程 1

要求判断是否可以完成所有课程。如果可以,返回 true;否则返回 false

初始思路

一开始想到的是把二维数组转成链表,然后判断链表里是否存在循环。

但这题不是普通链表问题。一个课程可能有多个后续课程,也可能被多个课程依赖,所以它本质上是一张 有向图,不是一条链表。

如果课程之间存在环,就说明有些课程互相依赖,无法完成所有课程。

例如:

复制代码
0 -> 1
1 -> 0

这就表示学习 0 之前要先学 1,学习 1 之前又要先学 0,形成了死循环。

解题思路

这题可以用 DFS 判断有向图中是否存在环。

先根据 prerequisites 建图:

复制代码
g[p[1]].add(p[0]);

这里的方向是:

复制代码
先修课 -> 后续课

也就是如果 p = [a, b],表示学 a 之前必须先学 b,所以建边 b -> a

然后用三色标记记录每个课程的访问状态:

  • 0:未访问。
  • 1:正在访问。
  • 2:已经访问完成。

DFS 过程中,如果遇到一个状态为 1 的节点,说明这个节点还在当前递归路径上,又被重新访问到了,所以存在环。

如果存在环,就无法完成所有课程,返回 false

为什么需要三色标记

这题不能只用一个简单的 visited

因为有向图判环时,需要区分两种状态:

  • 这个点以前访问过,并且已经确认它后面的路径没有环。
  • 这个点正在当前 DFS 路径中,还没有退出递归。

只有遇到"正在访问"的点,才说明形成了环。

也就是代码里的:

复制代码
if (color[y] == 1) {
    return true;
}

当一个节点的所有后续节点都 DFS 完成后,要把它标记成 2

复制代码
color[x] = 2;

表示这个点已经检查完成,以后再遇到它就不用重复搜索。

易错点

1. 建图方向

prerequisites[i] = [ai, bi] 的含义是:学 ai 之前要先学 bi

所以边的方向应该是:

复制代码
bi -> ai

对应代码:

复制代码
g[p[1]].add(p[0]);

2. DFS 结束后要标记为完成

如果一个点搜索完没有发现环,要把它从 1 改成 2

否则其他路径再次访问到它时,可能会误以为遇到了环。

3. 外层要遍历所有课程

图不一定是连通的。

有些课程可能和课程 0 完全不在一个连通块里,所以不能只从一个课程开始 DFS,而是要遍历所有课程:

复制代码
for (int i = 0; i < numCourses; i++) {
    if (color[i] == 0 && dfs(i, g, color)) {
        return false;
    }
}

代码实现

复制代码
class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        List<Integer>[] g = new ArrayList[numCourses];
        Arrays.setAll(g, i -> new ArrayList<>());
        int[] color = new int[numCourses];

        for (int[] p : prerequisites) {
            g[p[1]].add(p[0]);
        }

        for (int i = 0; i < numCourses; i++) {
            if (color[i] == 0 && dfs(i, g, color)) {
                return false;
            }
        }

        return true;
    }

    public boolean dfs(int x, List<Integer>[] g, int[] color) {
        color[x] = 1;

        for (int y : g[x]) {
            if (color[y] == 1 || color[y] == 0 && dfs(y, g, color)) {
                return true;
            }
        }

        color[x] = 2;
        return false;
    }
}

复杂度分析

  • 时间复杂度:O(numCourses + prerequisites.length)。每个课程节点和每条先修边最多被访问一次。
  • 空间复杂度:O(numCourses + prerequisites.length)。邻接表需要存储所有边,递归栈和颜色数组最多需要 O(numCourses)

复盘

这题的关键是把课程关系看成一张有向图,然后判断图里有没有环。

最开始想用链表判环,是因为抓住了"循环依赖"这个方向,但没有意识到课程关系不是一条链,而是可能一对多、多对一的图结构。

下次遇到类似题时,可以先检查三点:

  • 依赖关系能不能抽象成有向图。
  • 边方向是否是 先修课 -> 后续课
  • DFS 判环时是否区分了 未访问正在访问已完成 三种状态。
相关推荐
致Great2 小时前
Pi 的上下文压缩,到底是怎么工作的?
算法
watersink3 小时前
机器学习聚类算法
算法·机器学习·聚类
luj_17684 小时前
桥牌思维启示:系统设计的模块化架构
c语言·开发语言·c++·经验分享·算法
饼饼学习空间智能5 小时前
家庭服务机器人训练数据怎么积累?仿真、真实采集与持续学习的技术路线分析
人工智能·算法·机器学习
蛋先生DX7 小时前
你瘦不下来但大模型可以:量化原理了解一下
深度学习·算法·llm
牛阿大8 小时前
LQR算法
算法
(╹◡╹)8 小时前
18.剪枝
算法·机器学习·剪枝
Fa_Mian_Tuan9 小时前
图论基础|邻接矩阵超详细讲解(含无向/有向/带权图+完整可运行C语言代码)
c语言·数据结构·笔记·算法·图论
hanhahai9 小时前
指针与函数(函数指针与指针函数)
算法