[Leetcode] 207 & 210 课程表

你这个学期必须选修 numCourses 门课程，记为 0 到 numCourses - 1 。

在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出，其中 prerequisites[i] = [ai, bi] ，表示如果要学习课程 ai 则 必须 先学习课程  bi 。

例如，先修课程对 [0, 1] 表示：想要学习课程 0 ，你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习？如果可以，返回 true ；否则，返回 false 。

code：

public class CanFinish {

    // 207. 课程表
    public boolean canFinish(int numCourses, int[][] prerequisites) {

        int[] cnt = new int[numCourses];
        for (int[] pair : prerequisites) {
            cnt[pair[0]]++;
        }
        LinkedList<Integer> que = new LinkedList<>();
        for (int i = 0; i < numCourses; i++) {
            if (cnt[i] == 0) {
                que.addLast(i);
            }
        }
        int cntSize = que.size();
        while (!que.isEmpty()) {
            Integer x = que.removeFirst();
            for (int[] pair : prerequisites) {
                if (pair[1] == x) {
                    cnt[pair[0]]--;
                    if (cnt[pair[0]] == 0) {
                        que.addLast(pair[0]);
                        cntSize++;
                    }
                }
            }
        }
        return numCourses == cntSize;
    }

    // 210. 课程表 II
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        int[] cnt = new int[numCourses];
        for (int[] pair : prerequisites) {
            cnt[pair[0]]++;
        }
        LinkedList<Integer> que = new LinkedList<>();
        for (int i = 0; i < numCourses; i++) {
            if (cnt[i] == 0) {
                que.addLast(i);
            }
        }
        int cntSize = que.size();
        int[] result = new int[numCourses];
        int index = 0;
        while (!que.isEmpty()) {
            Integer x = que.removeFirst();
            result[index++] = x;
            for (int[] pair : prerequisites) {
                if (pair[1] == x) {
                    cnt[pair[0]]--;
                    if (cnt[pair[0]] == 0) {
                        que.addLast(pair[0]);
                        cntSize++;
                    }
                }
            }
        }
        if (numCourses != cntSize) {
            return new int[]{};
        }
        return result;
    }

    public static void main(String[] args) {
        int[][] prerequisites = new int[][]{{1, 0}};
        CanFinish solution = new CanFinish();
        System.out.println(solution.canFinish(2, prerequisites));
    }
}