Leetcode 1462. 课程表 IV (判断两个课程是否存在依赖，即等价于有向图的多源最短路径）

wwxy261

于 2021-10-23 00:27:56 发布

阅读量267

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/wwxy1995/article/details/120915628

版权

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

这篇文章介绍了如何使用优雅的Python代码，通过Floyd算法解决有向图中从顶点A到顶点B是否存在路径的问题。作者首先构建了邻接矩阵表示图，然后利用动态规划的思想进行迭代计算，最后返回每个查询点之间的可达性结果。

摘要由CSDN通过智能技术生成

开始以为是并查集，判断两个点是否联通，但是考虑到是有向图，实际上使用floyd算法判断是否可以从A点到B点，优雅的Python写法

class Solution:
    def checkIfPrerequisite(self, n: int, prerequisites: List[List[int]], queries: List[List[int]]) -> List[bool]:
        dp = [[False] * n for _ in range(n)]

        # every edge
        for u, v in prerequisites:
            dp[u][v] = True

        # dp里面装着dp[i][j][k-1]的结果
        for k in range(n):
            for i in range(n):
                for j in range(n):
                    dp[i][j] = dp[i][j] or (dp[i][k] and dp[k][j])

        return [dp[u][v] for u, v in queries]