给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
某个点能否有一个路径通往边缘,又是深度优先搜索的问题,思路依旧常规:一个主循环遍历和一个辅助的dfs函数,本题由于要判断两种边界,因此设置了两个搜索函数,一个搜索能到达太平洋的节点,一个搜索大西洋的。搜索函数的思路为:若该点坐标越界返回false,若该点到达海洋返回true,否则递归其四方元素,注意递归的条件是其值小于等于上一层递归的值。
这里有一个我错误了很久的点,最后一直没想出来才借鉴了别人的代码:进入递归前要把该点元素值变为-1,最后还原,来防止一直水流(即搜索)一直在某个区域内循环的问题,否则会导致一直递归而栈溢出。
class Solution {
public List<List<Integer>> pacificAtlantic(int[][] matrix) {
List<List<Integer>> res = new ArrayList<>();
if(matrix == null || matrix.length == 0){
return res;
}
for(int i=0; i<matrix.length; i++){
for(int j=0; j<matrix[0].length; j++){
if(dfsPacific(matrix,i,j,Long.MAX_VALUE)
&& dfsAtlantic(matrix,i,j,Long.MAX_VALUE)){
res.add(Arrays.asList(i,j));
}
}
}
return res;
}
//“太平洋”处于大陆的左边界和上边界
private boolean dfsPacific(int[][] matrix, int i, int j, long prev){
if(i > matrix.length-1 || j > matrix[0].length-1 || matrix[i][j] == -1 || matrix[i][j] > prev){
return false;
}
//到了太平洋的边界,说明可以流入,返回true
if(i <= 0 || j <= 0){
return true;
}
//临时记录当前matrix[i][j]的值,防止重复遍历
int temp = matrix[i][j];
matrix[i][j] = -1;
boolean result = dfsPacific(matrix,i-1,j,temp)
|| dfsPacific(matrix,i+1,j,temp)
|| dfsPacific(matrix,i,j-1,temp)
|| dfsPacific(matrix,i,j+1,temp);
matrix[i][j] = temp;
return result;
}
//“大西洋”处于大陆的右边界和下边界
private boolean dfsAtlantic(int[][] matrix, int i, int j, long prev){
if(i < 0 || j < 0 || matrix[i][j] == -1 || matrix[i][j] > prev){
return false;
}
//到了大西洋的边界,说明可以流入,返回true
if(i >= matrix.length-1 || j >= matrix[0].length-1){
return true;
}
//临时记录当前matrix[i][j]的值,防止重复遍历
int temp = matrix[i][j];
matrix[i][j] = -1;
boolean result = dfsAtlantic(matrix,i-1,j,temp)
|| dfsAtlantic(matrix,i+1,j,temp)
|| dfsAtlantic(matrix,i,j-1,temp)
|| dfsAtlantic(matrix,i,j+1,temp);
matrix[i][j] = temp;
return result;
}
}