python leetcode 417. Pacific Atlantic Water Flow

最新推荐文章于 2022-01-07 17:32:14 发布

原创最新推荐文章于 2022-01-07 17:32:14 发布 · 825 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文深入探讨了使用深度优先搜索（DFS）算法解决水流动模拟问题的方法，具体为判断二维矩阵中哪些单元格可以同时流向太平洋和大西洋。通过递归搜索策略，从矩阵边界开始遍历，确保每个可达单元格至少高于其相邻单元格，从而实现两大洋的水流动模拟。文章详细解释了如何设置搜索起始位置，并提供了完整的Python代码实现。

考察DFS，难点在于要分别判断是否能到达两大洋及如何设置搜索起始位置

class Solution:
    def pacificAtlantic(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[List[int]]
        """
        m=len(matrix)
        if m==0: return []
        n=len(matrix[0])
        Pdp=[[False]*n for _ in range(m)]
        Adp=[[False]*n for _ in range(m)]
        for i in range(m):
            self.dfs(Pdp,matrix,m,n,i,0)
            self.dfs(Adp,matrix,m,n,i,n-1)
        for j in range(n):
            self.dfs(Pdp,matrix,m,n,0,j)
            self.dfs(Adp,matrix,m,n,m-1,j)
        res=[]
        for i in range(m):
            for j in range(n):
                if Pdp[i][j] and Adp[i][j]:
                    res.append([i,j])
        return res
    def dfs(self,visit,matrix,m,n,x,y):
        visit[x][y]=True 
        for dx,dy in zip((-1,0,1,0),(0,-1,0,1)):
            nx,ny=x+dx,y+dy
            if nx<0 or nx>=m or ny<0 or ny>=n or visit[nx][ny] or matrix[nx][ny]<matrix[x][y]:
                continue 
            self.dfs(visit,matrix,m,n,nx,ny)