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

 

 

只要能走到第一行或者第一列，那么就能流入太平洋

能走到最后一列或者最后一行，那么就能流入大西洋

从第一行或者第一列出发，从小到大遍历，能遍历到的reachP=1

从最后一列或者最后一行出发，从小到大遍历，能遍历到的reachX=1

如果一个坐标的reachP=1并且reachX=1，说明它能流入太平洋和大西洋

 

class Solution {
private:
    vector<vector<int>> dir = {{0,1},{0,-1},{1,0},{-1,0}};
    void dfs(vector<vector<int>>& matrix, int x, int y, vector<vector<bool>> &reach)
    {
        if(reach[x][y]) return;
        reach[x][y] = true;
        int row = matrix.size();
        int col = matrix[0].size();
        for(int i = 0; i < 4; i++)
        {
            int nx = x + dir[i][0];
            int ny = y + dir[i][1];
            //如果下一个点超过范围或者比现在这个点大，不走
            if(nx < 0 || ny < 0 || nx >= row || ny >= col || matrix[nx][ny] < matrix[x][y]) continue;
            dfs(matrix, nx, ny, reach);
        }
    }
public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
        vector<vector<int>> res;
        int row = matrix.size();
        if(!row) return res;
        int col = matrix[0].size();

        vector<vector<bool> > reachP(row, vector<bool>(col, false));
        vector<vector<bool> > reachX(row, vector<bool>(col, false));
        
        //太平洋在第一行,大西洋在最后一行
        for(int i = 0; i < col; i++)
        {
            dfs(matrix, 0, i, reachP);
            dfs(matrix, row - 1, i, reachX);
        }
        //太平洋在第一列,大西洋在最后一列
        for(int i = 0; i < row; i++)
        {
            dfs(matrix, i, 0, reachP);
            dfs(matrix, i, col - 1, reachX);
        }
        for(int i = 0; i < row; i++)
        {
            for(int j = 0;j < col; j++)
            {
                if(reachP[i][j] && reachX[i][j])
                {
                    res.push_back({i, j});
                }
            }
        }
        return res;
    }
};