/*
题目:太平洋大西洋水流问题 middle
给定一个 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]] (上图中带括号的单元).
*/
/*
思路:
1.bfs,本题采用bfs
2.dfs
*/
package leetCode.BFS;
import java.util.*;
public class lc_bfs_417_pacificAtlantic {
/**
* 从边缘往内陆扩散,bfs
* 采用两个队列,分别用来代表Pacific和Atlantic的bfs
*
* @return
*/
public List<List<Integer>> pacificAtlantic(int[][] matrix) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
if (matrix == null || matrix.length == 0) {
return ans;
}
int m = matrix.length;
int n = matrix[0].length;
int[][] dir = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
ArrayDeque<int[]> qAtl = new ArrayDeque<>();
ArrayDeque<int[]> qPac = new ArrayDeque<>();
boolean[][] visAtl = new boolean[m][n];
boolean[][] visPac = new boolean[m][n];
//放在同一个双重for循环里,不要分开,简洁。
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
//Pacific边缘开始bfs
if (i == 0 || j == 0) {
visPac[i][j] = true;
qPac.offer(new int[]{i, j});
bfs(qPac, dir, visPac, matrix);
}
//Atlantic边缘开始bfs
if (i == m - 1 || j == n - 1) {
visAtl[i][j] = true;
qAtl.offer(new int[]{i, j});
bfs(qAtl, dir, visAtl, matrix);
}
}
}
//只有两个都访问过的才符合要求
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (visAtl[i][j] && visPac[i][j]) {
ans.add(Arrays.asList(i, j));
}
}
}
return ans;
}
public void bfs(Deque<int[]> q, int[][] dir, boolean[][] vis, int[][] matrix) {
while (!q.isEmpty()) {
int[] point = q.poll();
for (int k = 0; k < dir.length; k++) {
int x = point[0] + dir[k][0];
int y = point[1] + dir[k][1];
//不越界 且 没访问过 且 新陆地高度不小于原陆地 才继续扩散
if (x >= 0 && y >= 0 && x < matrix.length && y < matrix[0].length
&& !vis[x][y] && matrix[x][y] >= matrix[point[0]][point[1]]) {
q.offer(new int[]{x, y});
vis[x][y] = true;
}
}
}
}
}