太平洋大西洋水流问题

给定一个 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

采用回溯法会使问题变得简单,采用两个辅助矩阵sea1和sea2分别表示从太平洋和大西洋进行回溯能到达的区域,之后取两者交集即为答案。
对边界依次进行dfs回溯,完成辅助矩阵的初始化。dfs时候注意边界条件的判断以及已经初始化过的位置不要重复对其进行dfs,否则会导致无限月读从而StackOverflow。

class Solution {
      void dfs(int[][] matrix1,int[][] matrix2,int indexX,int indexY){
        if(indexX>=0&&indexY>=0&&indexX<matrix1.length&&indexY<matrix1[0].length){
            if(indexX!=0&&matrix1[indexX-1][indexY]>=matrix1[indexX][indexY]&&matrix2[indexX-1][indexY]!=1){
                matrix2[indexX-1][indexY]=1;
                dfs(matrix1,matrix2,indexX-1,indexY);
            }

            if(indexX!=matrix1.length-1&&matrix1[indexX+1][indexY]>=matrix1[indexX][indexY]&&matrix2[indexX+1][indexY]!=1){
                matrix2[indexX+1][indexY]=1;
                dfs(matrix1,matrix2,indexX+1,indexY);
            }

            if(indexY!=0&&matrix1[indexX][indexY-1]>=matrix1[indexX][indexY]&&matrix2[indexX][indexY-1]!=1){
                matrix2[indexX][indexY-1]=1;
                dfs(matrix1,matrix2,indexX,indexY-1);
            }

            if(indexY!=matrix1[0].length-1&&matrix1[indexX][indexY+1]>=matrix1[indexX][indexY]&&matrix2[indexX][indexY+1]!=1){
                matrix2[indexX][indexY+1]=1;
                dfs(matrix1,matrix2,indexX,indexY+1);
            }
        }
    }

    public List<List<Integer>> pacificAtlantic(int[][] matrix) {
        List<List<Integer>> ans=new ArrayList<>();
        if(matrix.length==0)
           return ans;
        int[][] sea1=new int[matrix.length][matrix[0].length];
        int[][] sea2=new int[matrix.length][matrix[0].length];

        for(int i=0;i<matrix[0].length;i++){
            sea1[0][i]=1;
            dfs(matrix,sea1,0,i);
            sea2[matrix.length-1][i]=1;
            dfs(matrix,sea2,matrix.length-1,i);
        }

        for(int i=0;i<matrix.length;i++){
            sea1[i][0]=1;
            dfs(matrix,sea1,i,0);
            sea2[i][matrix[0].length-1]=1;
            dfs(matrix,sea2,i,matrix[0].length-1);
        }

        for(int i=0;i<matrix.length;i++)
        {
            for(int j=0;j<matrix[0].length;j++)
            {
                if(sea1[i][j]==1&&sea2[i][j]==1)
                {
                    List<Integer> temp=new ArrayList<>();
                    temp.add(i);
                    temp.add(j);
                    ans.add(temp);
                }
            }
        }
        return ans;

    }
}
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容