Leetcode - Course Schedule II

原创于 2015-09-03 21:05:02 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构与算法

Graph 专栏收录该内容

4 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[分析]
在题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];
    }
}