417太平洋大西洋水流问题（中等题）

417太平洋大西洋水流问题（中等题）

题目

有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回 网格坐标 result 的 2D列表 ，其中 result[i] = [ri, ci] 表示雨水可以从单元格 (ri, ci) 流向 太平洋和大西洋 。
链接：https://leetcode-cn.com/problems/pacific-atlantic-water-flow
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

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

数据范围

m == heights.length
n == heights[r].length
1 <= m, n <= 200
0 <= heights[r][c] <= 105

题解

遍历每个单元格进行判断复杂度太高。
由于只需要判断是否能到达海洋，从海洋进行反推更快，具体可以看代码

class Solution {
public:
    int step[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};//定义可以水流走的方法
    vector<vector<int>>heights;
    void dfs(int r,int c, vector<vector<bool>>&o)//dfs进行判断边界的每个单元格是否到海洋
    {
        int m = this->heights.size(),n = this->heights[0].size();
        if(o[r][c])return;//如果能从这个单元格到达则返回
        o[r][c] = true;
        for(int i = 0; i < 4; i++)
        {
            int nr = r+step[i][0], nc = c + step[i][1];
            if(nr >= 0 && nr < m && nc >= 0 && nc < n && this->heights[r][c] <= this->heights[nr][nc])
                dfs(nr,nc,o);
        }
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        this->heights = heights;
        int m = heights.size(),n = heights[0].size();
        vector<vector<int>>ans;
        vector<vector<bool>>p(m, vector<bool>(n,false));
        vector<vector<bool>>a(m, vector<bool>(n,false));
        for(int i = 0; i < m; i++)dfs(i,0,p);//从两个海洋的边界进行dfs
        for(int j = 0; j < n; j++)dfs(0,j,p);
        for(int i = m-1; i >= 0; i--)dfs(i,n-1,a);
        for(int j = n-1; j >= 0; j--)dfs(m-1,j,a);
        for(int i = 0; i < m; i++)
        {
            for(int j = 0; j < n; j++)
            {
                if(p[i][j]&&a[i][j])
                {
                    vector<int>tmp;
                    tmp.push_back(i);
                    tmp.push_back(j);
                    ans.push_back(tmp);
                }
            }
        }
        return ans;
    }
};