[路飞]_leetcode 210. 课程表II

160 阅读2分钟

「这是我参与2022首次更文挑战的第6天,活动详情查看: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] 。

示例 3:

输入:numCourses = 1, prerequisites = [] 输出:[0]

分析

  • 首先初始化一个tableForNum数组长度为numCourses,并把内容填充为0。用于记录学习这门课有牵前置课的次数。没前置课说明课直接学。
let tableForNum = new Array(numCourses).fill(0)
prerequisites.forEach(nodeCouple => {
    let node0 = nodeCouple[0]
    tableForNum[node0]++
})
  • 定义queue队列,由于存放不需要前置课。
let queue = []
  • 定义findOrderArr数组,存放正确顺序。
let findOrderArr = []
  • 通过遍历tableForNum将不需要前置课的入队与queue
tableForNum.forEach((item, index) => {
    item === 0 && queue.push(index)
})
  • 现在就会出现俩种情况,一是没有个是不需要前置课的。出现这种情况是不可能完成的;另一种就是有不需要前置课的。

  • 遍历queue队列当queue长度等于0时结束循环

while (queue.length > 0) {

}
  • queue出队一个needItem,用于模拟学习一门不需要前置课的课程。每出队一个使numCourses减一。使其findOrderArr.push(needItem)
let needItem = queue.shift()
findOrderArr.push(needItem)
numCourses--
  • 通过循环prerequisites找出一门前置课是当前课程的,使他减1看是否等于0.如果等于0说明这门课的前置课已经学完了。并将其入队queue
for (let i = 0; i < prerequisites.length; i++) {
        let nodeCouple = prerequisites[i]
        findOrderArr.push(needItem)
        if (nodeCouple[1] != needItem) continue
        if (--tableForNum[nodeCouple[0]] == 0) {
            queue.push(nodeCouple[0])
        }
}
  • 通过者反复循环遍历,直至queue里没有值时结束
  • 最后通过判断numCourses是否等于0来判断课程表是否能被完成的如果完成了则返回findOrderArr,否则返回[]
return numCourses === 0 ? findOrderArr : []

题解

function canFinish(numCourses, prerequisites) {
    let tableForNum = new Array(numCourses).fill(0)
    prerequisites.forEach(nodeCouple => {
        let node0 = nodeCouple[0]
        tableForNum[node0]++
    })
    let queue = []
    tableForNum.forEach((item, index) => {
        item === 0 && queue.push(index)
    })
    while (queue.length > 0) {
        let needItem = queue.shift()
        findOrderArr.push(needItem)
        numCourses--
        for (let i = 0; i < prerequisites.length; i++) {
            let nodeCouple = prerequisites[i]
            if (nodeCouple[1] != needItem) continue
            if (--tableForNum[nodeCouple[0]] == 0) {
                queue.push(nodeCouple[0])
            }
        }
    }
    return  numCourses === 0 ? findOrderArr : []
}