[路飞]_课程表 II

142 阅读1分钟

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

210. 课程表 II

题目

现在你总共有 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]

题解

DFS

参考课程表

代码

var findOrder = function (numCourses, prerequisites) {
  const len = prerequisites.length

  // 出度,入度 的数组,数组表示当前课程的入度;入度为0,才可以学习;
  const outAndIn = Array(numCourses).fill(0)
  // 记录课程学完后,将课程相关的入度减小
  const keyValue = {}
  for (let i = 0; i < len; i++) {
    const [a, b] = prerequisites[i]
    outAndIn[a]++
    if (keyValue[b]) {
      keyValue[b].push(a)
    } else {
      keyValue[b] = [a]
    }
  }

  // 先找到课程入度为0的课
  const stack = []
  for (let i = 0; i < outAndIn.length; i++) {
    if (outAndIn[i] === 0) stack.push(i)
  }

  let result = []
  while (stack.length) {
    const t = stack.pop()
    result.push(t)
    const next = keyValue[t]
    if (next && next.length > 0) {
      for (let i = 0; i < next.length; i++) {
        outAndIn[next[i]] -= 1
        if (outAndIn[next[i]] === 0) {
          stack.push(next[i])
        }
      }
    }
  }
  return result.length === numCourses ? result : []
}

DFS