leetcode417 太平洋大西洋水流问题

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

思路：
使用搜索算法，建立两个与原矩阵相同大小的矩阵，分别记录能否到达太平洋和大西洋，然后分别从两个边界开始搜索，搜索周围的更大值，若搜索到则将其标记并继续搜索，若其已经被标记则直接返回，最后分别对比两个矩阵，若相同位置均被标记，则将其输出。

代码：

class Solution {
public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
int m=heights.size();
int n=heights[0].size();
vector<vector<bool>> w(m, vector<bool>(n, false));
vector<vector<bool>> e(m, vector<bool>(n, false));
for(int i=0;i<m;i++)
{
    dfs(heights,e,i,0);
    dfs(heights,w,i,n-1);
}
for(int i=0;i<n;i++)
{
    dfs(heights,e,0,i);
    dfs(heights,w,m-1,i);
}
vector<vector<int>> ans;
for (int i=0;i<m;i++) 
{
for (int j=0;j<n;j++) 
{
if(w[i][j]&&e[i][j]) 
{
ans.push_back(vector<int>{i, j});
} 
} 
}
return ans;
    }

    void dfs(vector<vector<int>>& h, vector<vector<bool>>& f,int p, int q)
    {
        vector<int> direction{-1, 0, 1, 0, -1};
        if(f[p][q]) return;
        else
        {
            f[p][q]=true;
            int x,y;
            for (int i=0;i<4;i++) 
            {
                x=p+direction[i];
                y=q+direction[i+1];
                if(x>=0&&x<h.size()&&y>=0&&y<h[0].size()&&h[p][q]<=h[x][y]) 
                {
                    dfs(h,f,x,y);
                    } 
                    }
        }
    }
};