lc_bfs_417_pacificAtlantic

/*
题目：太平洋大西洋水流问题 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;
                }
            }

        }

    }

}