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

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

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/pacific-atlantic-water-flow
先吐槽下力扣的翻译，真的有时候读了半天题目感觉莫名其妙，有时候会让人血压飙升。这道题目说白了就是给一个二维数组描述一个岛屿各个位置的高度，求雨水既能流入太平洋又能流入大西洋的位置。
思路分析：需要判断每个位置是否能够流入太平洋以及流入大西洋。可以从左边以及上边开始搜索，寻找哪些位置可以流向大西洋；接着从右边以及下边开始搜索，寻找哪些位置可以流向太平洋。我也使用深度优先算法进行代码编写，几乎和官方题解一样，不同的是我会对代码添加一些注释，方便读懂算法。
代码如下

class Solution {
    int[][] dirs = new int[][]{{0,-1},{0,1},{1,0},{-1,0}};//用于计算前后左右位置
    int row;//岛屿的行数
    int column;//岛屿的列数
    int[][] arr;//就是heights，只是方便函数调用
    public List<List<Integer>> pacificAtlantic(int[][] heights) {
        arr = heights;
        row = heights.length;
        column = heights[0].length;
        //创建是否可以流向两洋的布尔数组
        boolean[][] pacific = new boolean[row][column];
        boolean[][] atlantic = new boolean[row][column];
        //从左边界进行搜索
        for (int i = 0; i < row; i++) {
            dfs(i,0,pacific);
        }
        //从上边界进行搜索
        for (int j = 0; j < column; j++) {
            dfs(0,j,pacific);
        }
        //从右边界进行搜索
        for (int i = 0; i <row; i++) {
            dfs(i,column-1,atlantic);
        }
        //从下边界进行搜索
        for (int j = 0; j < column; j++) {
            dfs(row-1,j,atlantic);
        }
        //求数组pacific以及数组atlantic的交集
        //交集中的元素就是既能流向大西洋又能流向太平洋
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < column; j++) {
                if(pacific[i][j]&& atlantic[i][j]){
                    List<Integer> list = new ArrayList<>();
                    list.add(i);
                    list.add(j);
                    res.add(list);
                }
            }
        }

        return res;
    }
    //深度优先算法
    private void dfs(int i, int j, boolean[][] ocean) {
        //停止条件：如果待搜索的元素是true，那么说明已经搜索了该元素，可以停止
        if(ocean[i][j]){
            return;
        }
        //处理该元素，因为能够到达该元素，那么就该元素满足条件
        ocean[i][j] = true;
        //搜索相邻元素，上下左右
        for(int[] dir:dirs){
            int newRow = i+dir[0];
            int newColumn = j+dir[1];
            //相邻元素不能越界，前面的条件，最后一个条件就是找更高的位置，对应ocean[i][j] = true;语句进行理解
            if(newRow>=0 && newRow<row && newColumn>=0 && newColumn<column && arr[newRow][newColumn]>=arr[i][j]){
                dfs(newRow,newColumn,ocean);
            }
        }
    }
}