剑指 Offer II 113. 课程顺序(中等 图 bfs 拓扑排序 数组 哈希表)

剑指 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];
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值