给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
给定下面的 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,但是这里写这个博客的意义在于,目前深度搜索遇到的情况进行总结下:
1.普遍都是在一个二维数组中进行上下左右的搜索。
2.2个嵌套的for循环配上dfs函数,等修改判断标志位等操作。
3.但是,更精进点的是,搜索方式的改变,从上下左右到行遍历再dfs
4.这道题的思路改变,从矩形的外围操作,这是思路的技巧性转变,证明深度搜索这个思想,可以利用的地方很多,碰到复杂的题,更加要从反面的思考方式来思考问题。
class Solution {
private static final int[][] location = {{0,1},{0,-1},{-1,0},{1,0}};
private int n;
private int m;
public List<List<Integer>> pacificAtlantic(int[][] matrix) {
List<List<Integer>> list = new ArrayList<>();
List<Integer> tmp;
if(matrix == null || matrix.length == 0)
return list;
n = matrix.length;
m = matrix[0].length;
int[][] canP = new int[n][m];
int[][] canD = new int[n][m];
for(int i = 0;i < m;i++)
{
dfs(matrix,0,i,canP);
dfs(matrix,n-1,i,canD);
}
for(int j = 0;j < n;j++)
{
dfs(matrix,j,0,canP);
dfs(matrix,j,m-1,canD);
}
for(int i = 0;i<n;i++)
for(int j = 0;j<m;j++)
{
if(canP[i][j]==1&&canD[i][j]==1)
{
tmp = new ArrayList<>();
tmp.add(i);
tmp.add(j);
list.add(tmp);
}
}
return list;
}
public void dfs(int[][] matrix,int i,int j,int[][] can)
{
if(can[i][j]==1)
return;
can[i][j] = 1;
for(int[] k:location)
{
int nextR = i+k[0];
int nextC = j+k[1];
if(nextR<0 || nextC<0||nextR>=n||nextC>=m||matrix[i][j]>matrix[nextR][nextC])
continue;
dfs(matrix,nextR,nextC,can);
}
}
}