leetcode207. 课程表(dfs/bfs)

109 阅读1分钟

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

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

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

示例 1:

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

DFS代码

class Solution {
boolean[] finished;
    public boolean canFinish(int numCourses, int[][] prerequisites) {
     finished=new boolean[numCourses];
        HashMap<Integer,List<Integer>> map=new HashMap<>();//记录课程学完后可以学习的课程
        HashMap<Integer,HashSet<Integer>> map2=new HashMap<>();//记录需要先修的课程
        for(int i=0;i<numCourses;i++)//创建邻接表
        {
            map.put(i,new ArrayList<>());
            map2.put(i,new HashSet<>());
        }

        for(int[] pre:prerequisites)
        {
            map.get(pre[1]).add(pre[0]);
            map2.get(pre[0]).add(pre[1]);
        }
        for(int i=0;i<numCourses;i++)
            Finish(i,map,map2);
        for(int i=0;i<numCourses;i++)//检查是不是全部学完了
            if(!finished[i]) return false;
            return true;


    }
    public void Finish(int cur,HashMap<Integer,List<Integer>> map, HashMap<Integer,HashSet<Integer>> map2) {//深度优先搜索


        if(!map2.get(cur).isEmpty()) return;
        finished[cur]=true;
        for(int next:map.get(cur))
        {
            map2.get(next).remove(cur);
            Finish(next,map,map2);
        }


    }
}

BFS代码

    public boolean canFinish(int numCourses, int[][] prerequisites) {

        Queue<Integer> queue=new LinkedList<>();
        int[] adj=new int[numCourses];
        Map<Integer,Set<Integer>> pre=new HashMap<>();
        for(int i=0;i<numCourses;i++)
            pre.put(i,new HashSet<>());
        for(int[] t:prerequisites)
        {
            pre.get(t[1]).add(t[0]);//邻接表
            adj[t[0]]++;//记录节点的入度
        }

        for(int i=0;i<numCourses;i++)//入度为0的节点入队
            if(adj[i]==0) queue.add(i);
            while (!queue.isEmpty())
            {
                int cur=queue.poll();
                for(int c:pre.get(cur))//相邻节点入度减1
                {
                    if(--adj[c]==0) queue.offer(c);//入度为0的节点入队
                }

            }

        for(int i=0;i<numCourses;i++)
            if(adj[i]>0) return false;//还存在不能消除的边
            return true;

    }