给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
提示:
输出坐标的顺序不重要
m 和 n 都小于150
示例:
给定下面的 5x5 矩阵:
太平洋 ~ ~ ~ ~ ~
~ 1 2 2 3 (5) *
~ 3 2 3 (4) (4) *
~ 2 4 (5) 3 1 *
~ (6) (7) 1 4 5 *
~ (5) 1 1 2 4 *
* * * * * 大西洋
返回:
[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (上图中带括号的单元).
- 做法:dfs+记忆化搜索
- 我分两部分处理,第一个dfs处理能否到达太平洋,第二个dfs处理能否到达大西洋,我用path记录走过的路径,若为true
则可以到达,若为false则不可到达。(对每个点进行上下左右搜索)
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];
}
}