剑指 Offer II 113. 课程顺序
现在总共有 numCourses 门课需要选,记为 0 到 numCourses-1。
给定一个数组 prerequisites ,它的每一个元素 prerequisites[i] 表示两门课程之间的先修顺序。 例如 prerequisites[i] = [ai, bi] 表示想要学习课程 ai ,需要先完成课程 bi 。
请根据给出的总课程数 numCourses 和表示先修顺序的 prerequisites 得出一个可行的修课序列。
可能会有多个正确的顺序,只要任意返回一种就可以了。如果不可能完成所有课程,返回一个空数组。
示例 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,1,2,3] or [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 门课,直接修第一门课就可。
提示:
1 <= numCourses <= 2000
0 <= prerequisites.length <= numCourses * (numCourses - 1)
prerequisites[i].length == 2
0 <= ai, bi < numCourses
ai != bi
prerequisites 中不存在重复元素
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/QA2IGt
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
拓扑排序。所有的课程可以表示为一个有向无环图,需要先学习的课程指向后学习的课程,要得到课程顺序只需遍历整个图删除所有度为 0 的节点并输出即可(广度优先搜索)。
1、构建图。用哈希表记录每一个课程节点以及其指向的其他课程节点(邻接点),key 为一个课程节点,value 为一个动态数组记录邻接点,遍历 prerequisites 把所有的课程节点填到哈希表中,同时用一个数组记录所有课程节点的度(每个节点出现的次数),度为 0 说明课程节点只出现过一次,是一条路径的开头。
2、bfs 遍历图并记录课程学习顺序。广度优先搜索需要用到队列,队列初始化填入所有度为 0 的节点,接下来进行广度优先搜索,每次取出度为 0 的节点课程,并放入新的度为 0 的节点课程(取出度为 0 的节点课程会使其邻接点的度减 1)。所有取出的度为 0 的节点放入动态数组中。
3、如果存放度为 0 的节点动态数组的 size 小于课程数说明有些课程没学,输出 new int[0],否则输出该动态数组转化的静态数组。
题解(Java)
class Solution {
public int[] findOrder(int numCourses, int[][] prerequisites) {
Map<Integer, List<Integer>> graph = new HashMap<>();
for (int i = 0; i < numCourses; i++) {
graph.put(i, new LinkedList<Integer>());
}
int[] record = new int[numCourses];
for (int[] prerequisite : prerequisites) {
graph.get(prerequisite[1]).add(prerequisite[0]);
record[prerequisite[0]]++;
}
Queue<Integer> queue = new LinkedList<>();
for (int i = 0; i < numCourses; i++) {
if (record[i] == 0) {
queue.offer(i);
}
}
List<Integer> list = new LinkedList<>();
while (!queue.isEmpty()) {
int course = queue.poll();
list.add(course);
for (int g : graph.get(course)) {
record[g]--;
if (record[g] == 0) {
queue.offer(g);
}
}
}
return list.size() == numCourses ? list.stream().mapToInt(i -> i).toArray() : new int[0];
}
}