给定一个 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]] (上图中带括号的单元).
思路:
建立两个矩阵Atlantic和Pacific, 当Atlantic[i][j]和Pacific[i][j]同时为true时表示该位置可以同时到达Atlantic和Pacific
遍历时的技巧为: 只需要从四个边界开始遍历即可(类似泛洪的思想, 只要可以从边界出发到达, 就说明该位置的水可以流向对应的海洋)
建立两个boolean[][]类型的二维数组,只要能走通的路线都是true,最后只要找到这两个二维数组都是true的结点,就是通路的结点。
class Solution {
List<List<Integer>> list = new ArrayList();
int[][] directions = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
int row;
int col;
public List<List<Integer>> pacificAtlantic(int[][] matrix) {
if (matrix == null || matrix.length == 0) {
return list;
}
row = matrix.length;
col = matrix[0].length;
boolean[][] pacific = new boolean[row][col];
boolean[][] atlantic = new boolean[row][col];
for (int i = 0; i < row; i++) {
dfs(matrix, i, 0, pacific);
dfs(matrix, i, col - 1, atlantic);
}
for (int i = 0; i < col; i++) {
dfs(matrix, 0, i, pacific);
dfs(matrix, row - 1, i, atlantic);
}
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
if (pacific[i][j] && atlantic[i][j]) {
List<Integer> l = new ArrayList();
l.add(i);
l.add(j);
list.add(l);
}
}
}
return list;
}
public void dfs(int[][] matrix, int i, int j, boolean[][] visited) {
visited[i][j] = true;
for (int[] arr : directions) {
int nr = i + arr[0];
int nc = j + arr[1];
if (nr < 0 || nr >= row || nc < 0 || nc >= col || visited[nr][nc] || matrix[i][j] > matrix[nr][nc]) {
continue;
}
dfs(matrix, nr, nc, visited);
}
}
}