LeetCode-207-课程表

LeetCode-207-课程表

思路

其实就是图的拓扑排序，使用邻接表来存储，用队列进行BFS

代码

 public boolean canFinish(int numCourses, int[][] prerequisites) {
           
        List<List<Integer>> edges=new ArrayList<>();    // 存储有向图
        int[] indeg=new int[numCourses];                // 存储每个节点的入度 
        int index=0;                                  
        for (int i = 0; i < numCourses; ++i) {
            edges.add(new ArrayList<Integer>());
        }
        for (int[] info : prerequisites) {
            edges.get(info[1]).add(info[0]);
            indeg[info[0]]++;
        }

        Queue<Integer> queue = new LinkedList<Integer>();
        // 将所有入度为 0 的节点放入队列中
        for (int i = 0; i < numCourses; ++i) {
            if (indeg[i] == 0) queue.offer(i);
        }

        while (!queue.isEmpty()) { 
            int u = queue.poll();   // 从队首取出一个节点 
            index++;
            for (int v: edges.get(u)) {
                indeg[v]--;
                if (indeg[v] == 0) queue.offer(v);
            }
        }

        return index == numCourses;
    }