太平洋大西洋水流问题

题目描述

• 有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。
• 这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
• 岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
• 返回网格坐标 result 的 2D 列表 ，其中 result[i] = [ri, ci] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。

分析

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/pacific-atlantic-water-flow/solution/tai-ping-yang-da-xi-yang-shui-liu-wen-ti-sjk3/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

• 水流的方向是从高流到低，所以从一个单元格开始遍历则可以判断当前单元格是否可以流入海洋。从一个单元格开始，通过搜索的方法模拟雨水的流动，则可以判断雨水是否可以从该单元格流向海洋。
• 如果直接以每个单元格作为起点模拟雨水的流动，则会重复遍历每个单元格，导致时间复杂度过高。为了降低时间复杂度，可以从矩阵的边界开始反向搜索寻找雨水流向边界的单元格，反向搜索时，每次只能移动到高度相同或更大的单元格。
• 由于矩阵的左边界和上边界是太平洋，矩阵的右边界和下边界是大西洋，因此从矩阵的左边界和上边界开始反向搜索即可找到雨水流向太平洋的单元格，从矩阵的右边界和下边界开始反向搜索即可找到雨水流向大西洋的单元格。
• 可以使用深度优先搜索实现反向搜索，搜索过程中需要记录每个单元格是否可以从太平洋反向到达以及是否可以从大西洋反向到达。反向搜索结束之后，遍历每个网格，如果一个网格既可以从太平洋反向到达也可以从大西洋反向到达，则该网格满足太平洋和大西洋都可以到达，将该网格添加到答案中。

class Solution {
	
	public List<List<Integer>> pacificAtlantic(int[][] heights) {

        if (heights.length == 0) {
            return null;
        }
        int N = heights.length;
        int M = heights[0].length;
        boolean[][] pacific = new boolean[N][M];
        boolean[][] atlantic = new boolean[N][M];
        // 最上边的一条边
        for (int i = 0; i < M; i++) {
            process(0, i, pacific, heights);
        }
        for (int i = 0; i < N; i++) {
            process(i, 0, pacific, heights);
        }
        for (int i = 0; i < M; i++) {
            process(N - 1, i, atlantic, heights);
        }
        for (int i = 0; i < N; i++) {
            process(i, M - 1, atlantic, heights);
        }
        List<List<Integer>> result = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (pacific[i][j] && atlantic[i][j]) {
                    List<Integer> cur = new ArrayList<>();
                    cur.add(i);
                    cur.add(j);
                    result.add(cur);
                }
            }
        }
        return result;
    }

	// dfs
    int[][] choice = {{-1,0}, {0,1}, {1, 0}, {0, -1}}; // 四个方向
    private void process(int row, int col, boolean[][] grid, int[][] heights) {

        if (grid[row][col]) {
            return;
        }
		// 表明当前位置访问过
        grid[row][col] = true;
        for (int i = 0; i < 4; i++) {
            int newPointRow = row + choice[i][0];
            int newPointCol = col + choice[i][1];
			// 满足条件才进行访问
            if (newPointRow >= 0 && newPointRow < grid.length && newPointCol >= 0 && newPointCol < grid[0].length && heights[newPointRow][newPointCol] >= heights[row][col]) {
                process(newPointRow, newPointCol, grid, heights);
            }
        }
    }
}