力扣207、课程表 【图】

题目

课程与课程之间的先决条件组成有向图,课程是图中的顶点,课程之间的先决条件是图中的有向边。对于数组 prerequisites 中的元素 [a,b],表示在学习课程 a 前必须先完成课程 b,对应从 b 指向 a 的有向边,b 是 a 的前驱课程,a 是 b 的后继课程。

如果课程之间的先决条件不存在环,则可以完成所有课程;如果课程之间的先决条件存在环,则不能完成所有课程。因此判断是否可能完成所有课程等价于判断有向图是否为无环图。

解法

判断有向图是否为无环图可以使用拓扑排序,拓扑排序可以使用广度优先搜索或深度优先搜索实现。

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        List<Integer>[] adjacentArr = new List[numCourses];
        int[] indegrees = new int[numCourses];
        for (int i = 0; i < numCourses; i++) {
            adjacentArr[i] = new ArrayList<Integer>();
        }
        for (int[] prerequisite : prerequisites) {
            adjacentArr[prerequisite[1]].add(prerequisite[0]);
            indegrees[prerequisite[0]]++;
        }
        int finishCount = 0;
        Queue<Integer> queue = new ArrayDeque<Integer>();
        for (int i = 0; i < numCourses; i++) {
            if (indegrees[i] == 0) {
                queue.offer(i);
            }
        }
        while (!queue.isEmpty()) {
            int curr = queue.poll();
            finishCount++;
            List<Integer> adjacent = adjacentArr[curr];
            for (int next : adjacent) {
                indegrees[next]--;
                if (indegrees[next] == 0) {
                    queue.offer(next);
                }
            }
        }
        return finishCount == numCourses;
    }
}

你可能感兴趣的:(数据结构和算法,leetcode,算法)