leetcode207. 课程表(dfs/bfs)

你这个学期必须选修 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;

    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值