2020-12-22

题目描述：给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界，而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动，且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”，又能流动到“大西洋”的陆地单元的坐标。
解题思路：如果遍历整个矩阵中的所有位置，在不进行剪枝的情况下肯定会出现很多重复求解的过程，我们反过来思考，考虑从大西洋和太平洋出发往上流看可以到达哪些位置，两个都可以到达的位置就是我们的解，这样的话就只需要遍历矩阵的四条边即可，代码如下：

class Solution:
    def pacificAtlantic(self, matrix: List[List[int]]) -> List[List[int]]:
        m = len(matrix)
        if m == 0: return []
        n = len(matrix[0])
        if n == 0: return []
        reach_p = [[False]*n for _ in range(m)]
        reach_a = [[False]*n for _ in range(m)]
        direc = [0, -1, 0, 1, 0]
        def dfs(i, j, visited):
            if visited[i][j]: return
            visited[i][j] = True
            for k in range(4):
                tj = j + direc[k]
                ti = i + direc[k+1]
                if 0 <= tj < n and 0 <= ti < m and matrix[ti][tj] >= matrix[i][j] and not visited[ti][tj]:
                    dfs(ti, tj, visited)
        for i in range(n):
            dfs(0, i, reach_p)
            dfs(m-1, i, reach_a)
        for i in range(m):
            dfs(i, 0, reach_p)
            dfs(i, n-1, reach_a)
            
        res = []
        for i in range(m):
            for j in range(n):
                if reach_p[i][j] and reach_a[i][j]:
                    res.append([i, j])
        return res