LeetCode_207课程表

我是研究僧i 提交于 2019-12-13 16:42:13

现在你总共有 n 门课需要选,记为 0 到 n-1。

在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]

给定课程总量以及它们的先决条件,判断是否可能完成所有课程的学习?

示例 1:

输入: 2, [[1,0]]
输出: true
解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
示例 2:

输入: 2, [[1,0],[0,1]]
输出: false
解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成​课程 0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。
说明:

输入的先决条件是由边缘列表表示的图形,而不是邻接矩阵。详情请参见图的表示法。
你可以假定输入的先决条件中没有重复的边。
提示:

这个问题相当于查找一个循环是否存在于有向图中。如果存在循环,则不存在拓扑排序,因此不可能选取所有课程进行学习。
通过 DFS 进行拓扑排序 - 一个关于Coursera的精彩视频教程(21分钟),介绍拓扑排序的基本概念。
拓扑排序也可以通过 BFS 完成。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/course-schedule
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        int len1 = prerequisites.length;
        if(len1<1) return true;
        int len2 = prerequisites[0].length;
        if(len2<1) return true;
        //先构造有向图邻接矩阵
        boolean[][] graph = new boolean[numCourses][numCourses];
        for(int i=0;i<len1;i++){
            int col = prerequisites[i][0];
            int row = prerequisites[i][1];
            graph[row][col] = true;
        }
        //记录入度为0的点作为起点
        List<Integer> list = new ArrayList<Integer>();
        for(int j=0;j<numCourses;j++){
            boolean flag = true;
            for(int i=0;i<numCourses;i++){//判断入度为0
                if(graph[i][j]==true){
                    flag = false;
                    break;
                }
            }
            if(flag){
                list.add(j);
            }
        }
        if(list.size()<1){//不存在入度为0 的点
            return false;
        }
        //记录总的遍历过的节点
        Set<Integer> totalSet = new HashSet<Integer>();
        for(int i=0;i<list.size();i++){
             //记录这次遍历过的点
            Set<Integer> set = new HashSet<Integer>();
            boolean flag = search(graph,list.get(i),set,totalSet);
            if(flag==false){
                return false;
            }
        }
        if(totalSet.size()<numCourses){
            return false;
        }
        return true;
    }

    //DFS遍历
    public boolean search(boolean[][] graph, int row, Set<Integer> set, Set<Integer> totalSet){
        if(!totalSet.contains(row)){
            totalSet.add(row);
        }
        for(int j=0;j<graph.length;j++){
        	boolean flag = true;
            if(graph[row][j]==true){
                if(set.contains(row)){
                    return false;
                }else{
                    set.add(row);
                }
                flag = search(graph,j,set,totalSet);
                if(flag==false) {
                	return false;
                }
                set.remove(row);
            }
        }
        return true;
    }
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!