力扣 417. 太平洋大西洋水流问题

题目

有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。

这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。

岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。

返回网格坐标 result 的 2D 列表 ，其中 result[i] = [ri, ci] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。

示例

输入: heights = [[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]]

输入: heights = [[2,1],[1,2]]
输出: [[0,0],[0,1],[1,0],[1,1]]

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

方法1：DFS

⭐题解：https://leetcode-cn.com/problems/pacific-atlantic-water-flow/solution/shui-wang-gao-chu-liu-by-xiaohu9527-xxsx/

Java实现

class Solution {
    int m, n;
    int[] directions = new int[]{-1, 0, 1, 0, -1};
    int[][] heights;
    public List<List<Integer>> pacificAtlantic(int[][] heights) {
        this.heights = heights;
        m = heights.length;
        n = heights[0].length;
        boolean[][] visitedPo = new boolean[m][n];
        boolean[][] visitedAo = new boolean[m][n];

        for (int i = 0; i < m; i++) {
            track_back(i, 0, visitedPo);
            track_back(i, n - 1, visitedAo);
        }

        for (int j = 0; j < n; j++) {
            track_back(0, j, visitedPo);
            track_back(m - 1, j, visitedAo);
        }

        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (visitedPo[i][j] && visitedAo[i][j]) {
                    res.add(Arrays.asList(i, j));
                }
            }
        }

        return res;
    }

    public void track_back(int i, int j, boolean[][] visited) {
        if (visited[i][j]) return;

        visited[i][j] = true;
        for (int k = 0; k < directions.length - 1; k++) {
            int x = i + directions[k];
            int y = j + directions[k + 1];

            if (x < m && x >= 0 && y < n && y >= 0 && heights[x][y] >= heights[i][j]) {
                track_back(x, y, visited);
            }
        }
    }
}