leetcode 207. 课程表——java题解

题目所属分类

类似有向图的拓扑排序
入度为0就是起点
因为是要按照先后顺序的,所以是就是有向图

原题链接

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

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

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

代码案例:输入:numCourses = 2, prerequisites = [[1,0]]
输出:true
解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。

题解

leetcode 207. 课程表——java题解_第1张图片
1、由于课程有先后顺序,因此是一个拓扑排序问题,先建边,构图
2、走一遍拓扑排序,判断拓扑排序后点的个数cnt与总的课程n 是否相等
如果相等 那么说明无环 那么就能完成所有课程的学习

class Solution {
    public boolean canFinish(int n , int[][] prerequisites) {
        List<List<Integer>> g = new ArrayList<>();//邻接表
        int[] d = new int[n ];//入度
        for(int i = 0 ; i < n ; i ++ ){
            g.add(new ArrayList<>());
        }
        for(int[] p :prerequisites){
            int a = p[0] ;
            int b = p[1] ;
            g.get(a).add(b);
            d[b]++;
        }
        Queue<Integer> q = new LinkedList<>();
        for(int i = 0 ; i < n ; i++){
            if(d[i] == 0) q.offer(i);
        }
        int cnt = 0 ;//统计遍历的点数
        while(!q.isEmpty()){
            int t = q.poll();
            cnt++;
            for(int m : g.get(t)){
                if(--d[m] == 0){
                    q.offer(m);
                }
            }
        }
        return cnt == n ;
    }
}

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