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] 。

class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        Map<Integer, Integer> degree = new HashMap<>();// 记录入度
        ArrayList<Integer>[] arr = new ArrayList[numCourses];// 邻接表
        for (int i = 0; i < numCourses; i++) {
            degree.put(i, 0);// 初始化各个课程入度为0
            arr[i] = new ArrayList<Integer>();
        }

        for (int[] item : prerequisites) {
            degree.put(item[0], degree.get(item[0]) + 1);
            arr[item[1]].add(item[0]);
        }

        LinkedList<Integer> queue = new LinkedList<>();
        for (Map.Entry<Integer, Integer> entry : degree.entrySet()) {
            if (entry.getValue() == 0) {
                queue.offer(entry.getKey());
            }
        }

        LinkedList<Integer> res = new LinkedList<>();
        while (!queue.isEmpty()) {
            Integer poll = queue.poll();
            res.add(poll);
            for (Integer item : arr[poll]) {
                degree.put(item, degree.get(item) - 1);
                if (degree.get(item) == 0) {
                    queue.offer(item);
                }
            }
        }

        if (res.size() == numCourses) {
            int[] result = new int[numCourses];
            for (int i = 0; i < numCourses; i++) {
                result[i] = res.get(i);
            }
            return result;
        } else {
            return new int[] {};
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值