[路飞]_LeetCode_210. 课程表 II

「这是我参与2022首次更文挑战的第38天,活动详情查看:2022首次更文挑战

题目

现在你总共有 numCourses 门课需要选,记为 0 到 numCourses - 1。给你一个数组 prerequisites ,其中 prerequisites[i] = [ai, bi] ,表示在选修课程 ai 前 必须 先选修 bi 。

例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示:[0,1] 。 返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序,你只要返回 任意一种 就可以了。如果不可能完成所有课程,返回 一个空数组 。

示例 1:

输入:numCourses = 2, prerequisites = [[1,0]]
输出:[0,1]
解释:总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。
复制代码

示例 2:

输入:numCourses = 4, prerequisites = [[1,0],[2,0],[3,1],[3,2]]
输出:[0,2,1,3]
解释:总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。
并且课程 1 和课程 2 都应该排在课程 0 之后。
因此,一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3] 。
复制代码

示例 3:

输入:numCourses = 1, prerequisites = []
输出:[0]
复制代码

提示:

1 <= numCourses <= 2000
0 <= prerequisites.length <= numCourses * (numCourses - 1)
prerequisites[i].length == 2
0 <= ai, bi < numCourses
ai != bi
所有[ai, bi] 互不相同
复制代码

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/co… 著作权归领扣网络所有。

解题思路

这道题就是求有向图的拓扑序,如果序列长度等于课程总数,则返回拓扑序,否则返回空数组。

  • 计算每个节点的入度,并记录每个节点指向的节点;
  • 将入度为 0 的节点加入到队列;
  • 将队列里的元素依次出队列,同时将元素加入拓扑序。并且将出队列的节点指向的节点的入度减1,如果减1后人入度为0,则将该节点入队列;
  • 最后如果统计的个数和课程数相等则返回拓扑序,否则返回空数组;

代码实现

var findOrder = function(numCourses, prerequisites) {
    //用来存储入度
    const indeg = new Array(numCourses).fill(0)

    //每个点指向其它节点的集合
    const g = new Array(numCourses).fill(0).map(() => new Array())

    //队列用来存储入度为0的节点
    const q = new Array()

    for (const [curr, pre] of prerequisites) {
        // pre -> curr
        //当前节点的入度加1
        indeg[curr]++
        //前置节点指向当前节点
        g[pre].push(curr)
    }

    //把入度为0的节点入队列
    for (let i = 0; i < numCourses; i++) {
        if (indeg[i] === 0) q.push(i)
    }

    //统计有向图拓扑序节点个数
    let and = []
    while(q.length) {
        const ind = q.shift()
        and.push(ind)

        //把ind指向的节点的入度减一
        for (const i of g[ind]) {
            //如果i的入度为0则加入队列
            if (--indeg[i] === 0) q.push(i)
        }
    }

    //如果有向图节点个数等于课程数,则返回拓扑序,否则返回空数组
    return ans.length === numCourses ? ans : []
};
复制代码

如有错误欢迎指出,欢迎一起讨论!

猜你喜欢

转载自juejin.im/post/7068680808478801950