「这是我参与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 : []
};
如有错误欢迎指出,欢迎一起讨论!