【LeetCode笔记】207. 课程表（Java、图、BFS、队列）

题目描述

• 刷的第一道图相关的题，简直考古
• 要复习一下拓扑、邻接矩阵之类的知识啦
  

思路 && 代码

• 前置课程看成入度，后置课程看成出度
• 对于入度，只需要记录有多少即可。
• 对于出度，逐个记录下来（用于完成后，更新对应结点的入度）
• 初始化：从入度为0的结点开始记录
• BFS：可行课程加入队列中，先进先出地更新出入度。
• 时空复杂度：都是O(m + n)，结点和边都要遍历到

class Solution {
    // BFS，时空复杂度都是 O(M + N)，所有的邻边、结点都要遍历到
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        // 0. Init
        int[] indegrees = new int[numCourses]; // 入度记录
        Queue<Integer> queue = new LinkedList<>(); // BFS：当前可行结点记录
        List<List<Integer>> adjacency = new ArrayList<>(); // 出度结点记录
        for(int i = 0; i < numCourses; i++) {
            adjacency.add(new ArrayList<>());
        }

        // 1. 入度、出度记录
        for(int[] cp : prerequisites) {
            indegrees[cp[0]]++; // 入度计算
            adjacency.get(cp[1]).add(cp[0]); // 出度记录
        }

        // 2. 从入度为0的结点开始BFS
        for(int i = 0; i < numCourses; i++) {
            // 初始化：从入度为0的结点开始
            if(indegrees[i] == 0) {
                queue.add(i);
            }
        }

        // 3. 队列不为空时，每次移除一个队列课程，同时维护相关的出入度
        while(!queue.isEmpty()) {
            int pre = queue.poll();
            numCourses--;
            // 根据结点的出度列表，维护对应结点的入度值
            for(int temp : adjacency.get(pre)) {
                // 入度值为0时，加入可行结点的队列中
                if(--indegrees[temp] == 0) {
                    queue.add(temp);
                }
            }
        }
        return numCourses == 0;
    }
}

二刷

• 其实还是挺不错的一道题啦，步骤清晰、结构简单、思路明了，而且代码就19行
• 步骤分为三个部分：初始化、出入度初始化、课程维护

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        // Part 1: init
        int[] in = new int[numCourses]; 
        List<List<Integer>> out = new ArrayList<>(); 
        for(int i = 0; i < numCourses; i++) {
            out.add(new ArrayList<>());
        }

        // Part 2: 出入度维护
        for(int i = 0; i < prerequisites.length; i++) {
            in[prerequisites[i][0]]++;
            out.get(prerequisites[i][1]).add(prerequisites[i][0]);
        }

        // Part 3: 课程清除
        Deque<Integer> queue = new ArrayDeque<>(); 
        for(int i = 0; i < in.length; i++) {
            if(in[i] == 0) {
                queue.offer(i);
                numCourses--;
            }
        }
        while(!queue.isEmpty()) {
            int pre = queue.poll();
            for(int temp : out.get(pre)) {
                if(--in[temp] == 0) {
                    numCourses--;
                    queue.offer(temp);
                }
            }
        }
        return numCourses == 0;
    }
}