OJ练习第166题——课程表（拓扑排序问题）

课程表

力扣链接：207. 课程表

题目描述

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

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

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

示例

思路

拓扑排序问题，抓住节点入度和出度的本质特征。
方法一： 从入度思考(从前往后排序)， 入度为0的节点在拓扑排序中一定排在前面, 然后删除和该节点对应的边, 迭代寻找入度为0的节点。
方法二： 从出度思考(从后往前排序)， 出度为0的节点在拓扑排序中一定排在后面, 然后删除和该节点对应的边, 迭代寻找出度为0的节点。、

Java代码（从入度思考）

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        List<List<Integer>> edgs = new ArrayList<List<Integer>>();
        for(int i = 0; i < numCourses; i++) {
            edgs.add(new ArrayList<Integer>());
        }
        int[] indeg = new int[numCourses];
        for(int[] p : prerequisites) {
            edgs.get(p[1]).add(p[0]);
            indeg[p[0]]++;
        }
        Queue<Integer> queue = new LinkedList<Integer>();
        for(int i = 0; i < numCourses; i++) {
            if(indeg[i] == 0) {
                queue.offer(i);
            }
        }
        int visited = 0;
        while(!queue.isEmpty()) {
            visited++;
            int u = queue.poll();
            for(int v : edgs.get(u)) {
                indeg[v]--;
                if(indeg[v] == 0) {
                    queue.offer(v);
                }
            }
        }
        return visited ==numCourses;
    }
}

Java代码（从出度思考）

class Solution {
    List<List<Integer>> edgs;
    int[] visited;
    boolean valid = true;
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        edgs = new ArrayList<List<Integer>>();
        for(int i = 0; i < numCourses; i++) {
            edgs.add(new ArrayList<Integer>());
        }
        visited = new int[numCourses];
        for(int[] p : prerequisites) {
            edgs.get(p[1]).add(p[0]);
        }
        for(int i = 0; i < numCourses && valid; i++) {
            if(visited[i] == 0) {
                dfs(i);
            }
        }
        return valid;
    }
    public void dfs(int u) {
        visited[u] = 1;
        for(int v : edgs.get(u)) {
            if(visited[v] == 0) {
                dfs(v);
                if(!valid) {
                    return;
                }
            }else if(visited[v] == 1) {
                valid = false;
                return;
            }
        }
        visited[u] = 2;
    }
}

同类型题目

力扣链接：210. 课程表 II

题目描述

现在你总共有 numCourses 门课需要选，记为 0 到 numCourses - 1。给你一个数组 prerequisites ，其中 prerequisites[i] = [ai, bi] ，表示在选修课程 ai 前 必须 先选修 bi 。

例如，想要学习课程 0 ，你需要先完成课程 1 ，我们用一个匹配来表示：[0,1] 。
返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序，你只要返回 任意一种 就可以了。如果不可能完成所有课程，返回 一个空数组 。

示例

Java代码（从出度思考）

class Solution {
    List<List<Integer>> edges = new ArrayList<List<Integer>>();
    int[] indeg, result;
    int index;
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        for(int i = 0; i < numCourses; i++) {
            edges.add(new ArrayList<Integer>());
        }
        indeg = new int[numCourses];
        result = new int[numCourses];
        index = 0;
        for(int[] p : prerequisites) {
            edges.get(p[1]).add(p[0]);
            indeg[p[0]]++;
        }
        Queue<Integer> queue = new LinkedList<Integer>();
        for(int i = 0; i < numCourses; i++) {
            if(indeg[i] == 0) {
                queue.offer(i);
            }
        }
        while(!queue.isEmpty()) {
            int u = queue.poll();
            result[index++] = u;
            for(int v : edges.get(u)) {
                indeg[v]--;
                if(indeg[v] == 0) {
                    queue.offer(v);
                }
            }
        }
        if(index != numCourses) {
            return new int[0];
        }
        return result;
    }
}