安排课程

2024年6月7日 91次阅读

你需要去上n门九章的课才能获得offer，这些课被标号为 0 到 n-1 。
有一些课程需要“前置课程”，比如如果你要上课程0，你需要先学课程1，我们用一个匹配来表示他们： [0,1]

给你课程的总数量和一些前置课程的需求，返回你为了学完所有课程所安排的学习顺序。

可能会有多个正确的顺序，你只要返回一种就可以了。如果不可能完成所有课程，返回一个空数组。

样例
给定 n = 2, prerequisites = [[1,0]]
返回 [0,1]

给定 n = 4, prerequisites = [1,0],[2,0],[3,1],[3,2]]
返回 [0,1,2,3] or [0,2,1,3]

public class Solution {
    /** * @param numCourses a total of n courses * @param prerequisites a list of prerequisite pairs * @return the course order */
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        List[] edges = new ArrayList[numCourses];
        int[] degree = new int[numCourses];
        for (int i = 0; i < numCourses; i++) {
            edges[i] = new ArrayList<Integer>();
        }
        //计算入度，并保存有向边
        for (int i = 0; i < prerequisites.length; i++) {
            degree[prerequisites[i][0]]++;
            edges[prerequisites[i][1]].add(prerequisites[i][0]);
        }
        Queue queue = new LinkedList();
        int[] orders = new int[numCourses];
        int count = 0;
        //取出入度为0的点
        for (int i = 0; i < degree.length; i++) {
            if (degree[i] == 0) {
                queue.offer(i);
            }
        }
        //BFS遍历入度为0的点
        while (!queue.isEmpty()) {
            int num = (int) queue.poll();
            orders[count] = num;
            count++;
            for (int i = 0; i < edges[num].size(); i++) {
                int index = (int) edges[num].get(i);
                degree[index]--;
                if (degree[index] == 0) {
                    queue.offer(index);
                }
            }
        }
        if (count == numCourses) {
            return orders;
        }
        return new int[0];
    }
}
点赞