Leetcode 417. 太平洋大西洋水流问题 C++

最新推荐文章于 2025-07-20 15:38:46 发布

原创最新推荐文章于 2025-07-20 15:38:46 发布 · 445 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

本文详细解析了LeetCode上的经典题目——417太平洋大西洋水流问题，通过深度优先搜索（DFS）算法，寻找能够同时流向太平洋和大西洋的陆地单元坐标。代码示例清晰，帮助理解如何标记并追踪水流路径。

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

题解

dfs深搜
我们不妨把水看成从低处往高处流，这样我们从太平洋或大西洋开始流，就能得出哪些地方能留到太平洋或大西洋。我们最初从太平洋开始流，将流过的地方进行标记；然后再从大西洋开始流，如果流过的地方是之前进行标记的则放入答案中。详细过程见代码

代码

vector<vector<int>> ans;
    vector<vector<int>> use;
    void dfs(int x,int y,int m,int n,int choice,vector<vector<int>>& matrix){
        if(use[x][y] + choice == 3) ans.push_back({x,y});
        use[x][y] = choice;
        if(x-1>=0 && use[x-1][y]!=choice && matrix[x-1][y]>=matrix[x][y])	
            dfs(x-1,y,m,n,choice,matrix);
        if(x+1<m && use[x+1][y]!=choice && matrix[x+1][y]>=matrix[x][y])
            dfs(x+1,y,m,n,choice,matrix);
        if(y-1>=0 && use[x][y-1]!=choice && matrix[x][y-1]>=matrix[x][y])
            dfs(x,y-1,m,n,choice,matrix);
        if(y+1<n && use[x][y+1]!=choice && matrix[x][y+1]>=matrix[x][y])
            dfs(x,y+1,m,n,choice,matrix);
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
        if(matrix.empty() || matrix[0].empty()) return ans;
        int m = matrix.size(), n = matrix[0].size();
        use = vector(m,vector<int>(n,0));
        for(int i=0; i<m; i++)
            if(use[i][0]!=1)    dfs(i,0,m,n,1,matrix);
        for(int i=0; i<n; i++)
            if(use[0][i]!=1)    dfs(0,i,m,n,1,matrix);
        for(int i=0; i<m; i++)
            if(use[i][n-1]!=2)    dfs(i,n-1,m,n,2,matrix);
        for(int i=0; i<n; i++)
            if(use[m-1][i]!=2)  dfs(m-1,i,m,n,2,matrix);
        return ans;
    }

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