1. 程式人生 > 其它 >LeetCode-210. 課程表 II

LeetCode-210. 課程表 II

題目來源

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

示例 2:

輸入: numCourses = 4, prerequisites = [[1,0],[2,0],[3,1],[3,2]]
輸出: [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 <= numCourses <= 2000
  • 0 <= prerequisites.length <= numCourses * (numCourses - 1)
  • prerequisites[i].length == 2
  • 0 <= ai, bi < numCourses
  • ai != bi
  • 所有[ai, bi] 互不相同

相似題目

  1. 207. 課程表
  2. 210. 課程表 II

題解分析

解法一:拓撲排序

class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        List<List<Integer>> edges = new ArrayList<>();
        for(int i=0; i<numCourses; i++){
            edges.add(new ArrayList<>());
        }

        int[] incidient = new int[numCourses];// 入度矩陣
        for(int[] prerequisite : prerequisites){
            int from = prerequisite[1];
            int to = prerequisite[0];
            incidient[to]++;
            edges.get(from).add(to);// 構建鄰接表
        }

        Queue<Integer> que = new LinkedList<>();
        // 將入度為0的頂點入佇列
        for(int i=0; i<numCourses; i++){
            if(incidient[i] == 0){
                que.offer(i);
            }
        }
        int[] res = new int[numCourses];
        int cnt = 0;
        while(!que.isEmpty()){
            int v = que.poll();
            res[cnt] = v;
            cnt++;// 已經確定順序的頂點數
            for(int to : edges.get(v)){
                incidient[to]--;
                if(incidient[to] == 0){
                    que.offer(to);
                }
            }
        }
        return numCourses == cnt ? res : new int[]{};
    }
}