【图】【拓扑排序】leetcode207.课程表

461 篇文章 1 订阅

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

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

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

解答:
方法一:入读表(广度优先遍历BFS)
时间复杂度:O(N+M)
空间复杂度:O(N+M)
注: N 和 M 分别为节点数量和临边数量

class Solution:
    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        #课程安排图是否是 有向无环图(DAG):拓扑排序
        #统计课程安排图中每个节点的入度,生成入度表indegrees和邻接表
        indegrees=[0]*numCourses
        adj=defaultdict(list)
        for start,end in prerequisites:
            indegrees[end]+=1
            adj[start].append(end)
        #借助一个队列 queue,将所有入度为0的节点入队
        que=collections.deque()
        for v,degree in enumerate(indegrees):
            if degree==0:
                que.append(v)
        #当 queue非空时,依次将队首节点出队,在课程安排图中删除此节点 pre
        while que:
            cur=que.pop()
            numCourses-=1
            #将cur邻接节点的入度减1
            for end in adj[cur]:
                indegrees[end]-=1
                #若入度为0,则将邻接节点入队
                if indegrees[end]==0:
                    que.append(end)
        return numCourses==0

方法二:深度优先遍历
在这里插入图片描述

class Solution:
    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        #通过DFS判断图中是否有环
        #0.统计邻接表
        adj=defaultdict(list)
        for start,end in prerequisites:
            adj[start].append(end)
        #1.借助一个标志列表 flags,用于判断每个节点 i (课程)的状态:
        flags=[0]*numCourses
        #(i)未被 DFS 访问:i == 0;
        #(ii)已被其他节点启动的 DFS 访问:i == -1;
        #(iii)已被当前节点启动的 DFS 访问:i == 1。
        #2.对numCourses个节点依次执行DFS,判断每个节点起步DFS是否存在环,若存在环直接返回False。
        def dfs(cur):
            if flags[cur]==-1:
                return True 
            elif flags[cur]==1:
                return False
            #若当前节点cur还没被访问过
            flags[cur]=1
            for end in adj[cur]:
                if not dfs(end):
                    return False
            flags[cur]=-1
            return True
            
        #3.若整个图DFS结束并未发现环,返回True
        for v in range(numCourses):
            if not dfs(v):
                return False
        return True
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值