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] <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/pacific-atlantic-water-flow
 

方法一：

深度优先搜索，从外向内，通过每条边向内搜索可以流入的格子。

class Solution {
public:
    vector<vector<int>> heights;
    void dfs(int row,int col,vector<vector<bool>> &ocean){
        int m=heights.size(),n=heights[0].size();
        if(ocean[row][col]) return;
        ocean[row][col]=true;
        int dirs[4][2]={{-1,0},{1,0},{0,1},{0,-1}};
        for(int i=0;i<4;i++){
            int x=row+dirs[i][0];
            int y=col+dirs[i][1];
            if(x>=0&&x<m&&y>=0&&y<n&&heights[row][col]<=heights[x][y]){
                dfs(x,y,ocean);
            }
        }
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        this->heights=heights;
        int m=heights.size(),n=heights[0].size();
        vector<vector<bool>> pacific(m,vector<bool>(n,false));
        vector<vector<bool>> atlantic(m,vector<bool>(n,false));
        for(int i=0;i<n;i++){
            dfs(0,i,pacific);
        }
        for(int i=0;i<m;i++){
            dfs(i,0,pacific);
        }
        for(int i=0;i<m;i++){
            dfs(i,n-1,atlantic);
        }
        for(int i=0;i<n;i++){
            dfs(m-1,i,atlantic);
        }
        vector<vector<int>> ans;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(pacific[i][j]&&atlantic[i][j]){
                    ans.push_back({i,j});
                }
            }
        }
        return ans;
    }
};