题目
题目来源:力扣(LeetCode)
现在你总共有 n 门课需要选,记为 0 到 n-1。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]
给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。
可能会有多个正确的顺序,你只要返回一种就可以了。如果不可能完成所有课程,返回一个空数组。
示例 1:
输入: 2, [[1,0]]
输出: [0,1]
解释: 总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。
示例 2:
输入: 4, [[1,0],[2,0],[3,1],[3,2]]
输出: [0,1,2,3] or [0,2,1,3]
解释: 总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。
因此,一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3] 。
思路分析
拓补排序
给定一个包含 n 个节点的有向图 G,我们给出它的节点编号的一种排列,如果满足:
对于图 G 中的任意一条有向边 (u,v),u 在排列中都出现在 v 的前面。
那么称该排列是图 G 的「拓扑排序」。根据上述的定义,我们可以得出两个结论:
- 如果图 G 中存在环(即图 G 不是「有向无环图」),那么图 G 不存在拓扑排序。这是因为假设图中存在环 x1, x2, …, xn, x1,那么 x1 在排列中必须出现在 xn 的前面,但 xn 同时也必须出现在 x1 的前面,因此不存在一个满足要求的排列,也就不存在拓扑排序;
如果图 G 是有向无环图,那么它的拓扑排序可能不止一种。举一个最极端的例子,如果图 G 值包含 n 个节点却没有任何边,那么任意一种编号的排列都可以作为拓扑排序。
「拓扑排序」是专门应用于有向图的算法;
- BFS 的写法就叫「拓扑排序」,这里还用到了贪心算法的思想,贪的点是:当前让入度为 0 的那些结点入队;
- 「拓扑排序」的结果不唯一;
- 删除结点的操作,通过「入度数组」体现,这个技巧要掌握;
- 「拓扑排序」的一个附加效果是:能够顺带检测有向图中是否存在环,这个知识点非常重要,如果在面试的过程中遇到这个问题,要把这一点说出来。
- 具有类似附加功能的算法还有:Bellman-Ford 算法附加的作用是可以用于检测是否有负权环
有了上述的简单分析,我们就可以将本题建模成一个求拓扑排序的问题了:
- 我们将每一门课看成一个节点;
- 如果想要学习课程 A 之前必须完成课程 B,那么我们从 B 到 A 连接一条有向边。这样以来,在拓扑排序中,B 一定出现在 A 的前面。
算法流程:
- 在开始排序前,扫描对应的存储空间(使用邻接表),将入度为 0 的结点放入队列。
- 只要队列非空,就从队首取出入度为 0 的结点,将这个结点输出到结果集中,并且将这个结点的所有邻接结点(它指向的结点)的入度减 1,在减 1 以后,如果这个被减 1 的结点的入度为 0 ,就继续入队。
- 当队列为空的时候,检查结果集中的顶点个数是否和课程数相等即可。
思考这里为什么要使用队列?(马上就会给出答案。)
在代码具体实现的时候,除了保存入度为 0 的队列,我们还需要两个辅助的数据结构:
- 邻接表:通过结点的索引,我们能够得到这个结点的后继结点;
- 入度数组:通过结点的索引,我们能够得到指向这个结点的结点个数。
这个两个数据结构在遍历题目给出的邻边以后就可以很方便地得到。
/**
* @param {number} numCourses
* @param {number[][]} prerequisites
* @return {number[]}
*/
var findOrder = function (numCourses, prerequisites) {
const inDegree = new Array(numCourses).fill(0);//入度数组
const map = {};//邻接表
for (let i = 0; i < prerequisites.length; i++) {
inDegree[prerequisites[i][0]]++;//求每一堂课的初始入度值
if (map[prerequisites[i][1]]) {//当前课已经存在邻街表
map[prerequisites[i][1]].push(prerequisites[i][0])//添加依赖它的后续课
} else {//当前课不存在邻街表
map[prerequisites[i][1]] = [prerequisites[i][0]]
}
}
const queue = [];//处理入度为0节点存到队列
for (let i = 0; i < inDegree.length; i++) {//队列的初始化
if (inDegree[i] == 0) queue.push(i);
}
let ans = [];
while (queue.length) {
const selected = queue.shift();
// cnt++;
ans.push(selected);
const toEnQueue = map[selected];//当前课程的后续课程
if (toEnQueue && toEnQueue.length) {
for (let i = 0; i < toEnQueue.length; i++) {
inDegree[toEnQueue[i]]--;
if (inDegree[toEnQueue[i]] == 0) {
queue.push(toEnQueue[i])
}
}
}
}
if (ans.length == numCourses) return ans;
return [];
}