目录

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

解析代码

---

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

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

难度 中等

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

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

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

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

示例 1：

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

示例 2：

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

提示：

• m == heights.length
• n == heights[r].length
• 1 <= m, n <= 200
• 0 <= heights[r][c] <= 10^5

class Solution {
public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {

    }
};

---

解析代码

算法思路：正难则反。

        如果直接去判断某一个位置是否既能到大西洋也能到太平洋，会重复遍历很多路径。我们反着来，从大西洋沿岸开始反向 dfs ，这样就能找出那些点可以流向大西洋。同理，从太平洋沿岸也反向 dfs ，这样就能找出那些点可以流向太平洋。那么被标记两次的点，就是要找的结果。

class Solution {
    int dx[4] = {0, 0, -1, 1};
    int dy[4] = {1, -1, 0, 0};
    int m = 0, n = 0;

public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        m = heights.size(), n = heights[0].size();
        vector<vector<bool>> vis1(m, vector<bool>(n, false));
        vector<vector<bool>> vis2(m, vector<bool>(n, false));
        for(int i = 0; i < n; ++i) // 遍历周围
        {
            dfs(heights, 0, i, vis1); // 太平洋
            dfs(heights, m - 1, i, vis2); // 大西洋
        }
        for(int i = 0; i < m; ++i)
        {
            dfs(heights, i, 0, vis1); // 太平洋
            dfs(heights, i, n - 1, vis2); // 大西洋
        }
        vector<vector<int>> ret;
        for(int i = 0; i < m; i++)
        {
            for(int j = 0; j < n; j++)
            {
                if(vis1[i][j] && vis2[i][j])
                    ret.push_back({i, j});
            }
        }
        return ret;
    }

    void dfs(vector<vector<int>>& heights, int sr, int sc, vector<vector<bool>>& vis)
    {
        vis[sr][sc] = true;
        for(int i = 0; i < 4; ++i)
        {
            int x = sr + dx[i], y = sc + dy[i];
            if(x >= 0 && x < m && y >= 0 && y < n  && !vis[x][y] && heights[x][y] >= heights[sr][sc])
            {
                dfs(heights, x, y, vis);
            }
        }
    }
};