Leetcode - Course Schedule II

改进拓扑排序算法实现课程先修顺序

最新推荐文章于 2025-11-27 14:26:45 发布

原创最新推荐文章于 2025-11-27 14:26:45 发布 · 141 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构与算法

Graph 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种改进的拓扑排序算法，用于解决课程先修顺序问题。通过构建图并计算入度，算法实现了有效的课程排列，确保了学习路径的正确性。

[分析]
在题I代码基础上增加记录拓扑遍历顺序即可。


public class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        if (prerequisites == null)
            return new int[0];
        int[] order = new int[numCourses];
        List<Set<Integer>> graph = new ArrayList<Set<Integer>>();
        int[] indegree = new int[numCourses];
        for (int i = 0; i < numCourses; i++)
            graph.add(new HashSet<Integer>());
        // build graph
        for (int i = 0; i < prerequisites.length; i++) {
            if (graph.get(prerequisites[i][1]).add(prerequisites[i][0])) 
                indegree[prerequisites[i][0]]++;
        }
        LinkedList<Integer> queue = new LinkedList<Integer>();
        for (int i = 0; i < numCourses; i++) {
            if (indegree[i] == 0)
                queue.offer(i);
        }
        int k = 0;
        while (!queue.isEmpty()) {
            int v = queue.poll();
            order[k++] = v;
            for (Integer w : graph.get(v)) {
                if (--indegree[w] == 0)
                    queue.offer(w);
            }
        }
        return k == numCourses ? order : new int[0];
    }
}