Leetcode 417.太平洋大西洋水流问题

太平洋大西洋水流问题

给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。"太平洋"处于大陆的左边界和上边界，而"大西洋"处于大陆的右边界和下边界。

规定水流只能按照上、下、左、右四个方向流动，且只能从高到低或者在同等高度上流动。

请找出那些水流既可以流动到"太平洋"，又能流动到"大西洋"的陆地单元的坐标。

   

提示：

1. 输出坐标的顺序不重要
2. m 和 n 都小于150

   

示例：

   

给定下面的 5x5 矩阵:

 

返回:

 

[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (上图中带括号的单元).

 

 

题解

• 做法:dfs+记忆化搜索
• 分两部分处理，第一个dfs处理能否到达太平洋，第二个dfs处理能否到达大西洋，我用path记录走过的路径，若为true
  则可以到达，若为false则不可到达。(对每个点进行上下左右搜索)

 

import java.util.ArrayList;
import java.util.List;

class Solution {
    public boolean[][] vis;
    public boolean[][] vis1;
    public int n,m;
    public List<int[]> pacificAtlantic(int[][] matrix) {
        if(matrix.length==0||matrix[0].length==0){
            return new ArrayList<int[]>();
        }
        n=matrix.length;
        m=matrix[0].length;
        boolean[][] path=new boolean[n][m];
        boolean[][] path1=new boolean[n][m];
        vis=new boolean[n][m];
        vis1=new boolean[n][m];
        List<int[]> ans=new ArrayList<int[]>();
        for(int i = 0; i<n; ++i){
            for(int j = 0; j<m; ++j){
                if(dfs(i,j,path,matrix,matrix[i][j])&&dfs1(i,j,path1,matrix,matrix[i][j])){
                    ans.add(new int[]{i,j});
                }
            }
        }
        return ans;
    }
    public boolean dfs(int x, int y, boolean[][] path, int[][] matrix, int pre){
        if(x>=n||y>=m){
            return false;
        }
        if(x<0||y<0){
            return true;
        }
        if(matrix[x][y]>pre){
            return false;
        }
        if(vis[x][y]){
            return path[x][y];
        }
        vis[x][y]=true;
        path[x][y]=dfs(x-1,y,path,matrix,matrix[x][y])||dfs(x,y-1,path,matrix,matrix[x][y]) ||dfs(x+1,y,path,matrix,matrix[x][y])||dfs(x,y+1,path,matrix,matrix[x][y]);
        vis[x][y]=false;
        return path[x][y];
    }
    public boolean dfs1(int x,int y,boolean[][] path,int[][] matrix,int pre){
        if(x<0||y<0){
            return false;
        }
        if(x>=n||y>=m){
            return true;
        }
        if(matrix[x][y]>pre){
            return false;
        }
        if(vis1[x][y]){
            return path[x][y];
        }
        vis1[x][y]=true;
        path[x][y]=dfs1(x-1,y,path,matrix,matrix[x][y])||dfs1(x,y-1,path,matrix,matrix[x][y]) ||dfs1(x+1,y,path,matrix,matrix[x][y])||dfs1(x,y+1,path,matrix,matrix[x][y]);
        vis1[x][y]=false;
        return path[x][y];
    }
}