leetcode 417. 太平洋大西洋水流问题

给定一个 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]] (上图中带括号的单元).

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pacific-atlantic-water-flow
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

从矩阵的四边向反方向遍历。

定义两个布尔型二维数组b1,b2来记录是否能到达太平洋和大西洋。

递归遍历,若当前位置,符合要求,则向四边遍历,直到找不到合适的,再遍历下一个坐标节点。

    private int aa;
    private int bb;
    private int[][] heights;

    public List<List<Integer>> pacificAtlantic(int[][] heights) {
        List<List<Integer>> list = new ArrayList<>();
        int a = heights.length;
        if (a == 0) {
            return list;
        }
        int b = heights[0].length;
        if (b == 0) {
            return list;
        }
        aa = a - 1;
        bb = b - 1;
        this.heights = heights;
        boolean[][] b1 = new boolean[a][b];
        boolean[][] b2 = new boolean[a][b];
        for (int i = 0; i < a; i++) {
            b1[i][0] = true;
            b2[i][bb] = true;
        }
        for (int i = 0; i < b; i++) {
            b1[0][i] = true;
            b2[aa][i] = true;
        }
        for (int i = 0; i < a; i++) {
            find(b1, i, 1, heights[i][0]);
            find(b2, i, b - 2, heights[i][bb]);
        }
        for (int i = 0; i < b; i++) {
            find(b1, 1, i, heights[0][i]);
            find(b2, a - 2, i, heights[aa][i]);
        }

        for (int i = 0; i < a; i++) {
            for (int j = 0; j < b; j++) {
                if (b1[i][j] && b2[i][j]) {
                    List<Integer> item = new ArrayList<>(2);
                    item.add(i);
                    item.add(j);
                    list.add(item);
                }
            }
        }
        return list;
    }


    private void find(boolean[][] arr, int x, int y, int value) {
        if (x < 0 || x > aa || y < 0 || y > bb) {
            return;
        }
        if (arr[x][y]) {
            return;
        }
        int v = heights[x][y];
        if (v < value) {
            return;
        }
        arr[x][y] = true;
        find(arr, x - 1, y, v);
        find(arr, x + 1, y, v);
        find(arr, x, y - 1, v);
        find(arr, x, y + 1, v);
    }

posted @ 2021-06-21 10:43  旺仔古李  阅读(83)  评论(0编辑  收藏  举报