一、题目

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


二、思路

  • 就求某个点是否能到左上边界和右下边界。正向流动的话不太好实现，那么可以试着反过来，即从边界往中间流动。可以知道的是边界必然是能够到达一遍海洋的，那么就从边界的点进行搜索，看是否能到达内部的点，能到达即满足要求。那么求出所有满足2个边界都能到到达的点即可。

三、代码

class Solution {
public:
    vector<int> d{1, 0, -1, 0, 1};
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        int n = heights.size(), m = heights[0].size();
        vector<vector<int>> v1(n, vector<int>(m));
        vector<vector<int>> v2(n, vector<int>(m));
        vector<vector<int>> ans;
       for (int i = 0; i < n; i++) {
           bfs(heights, v1, i, 0);
           bfs(heights, v2, i, m - 1);
       }
       for (int i = 0; i < m; i++) {
           bfs(heights, v1, 0, i);
           bfs(heights, v2, n - 1, i);
       } 
        for (int i = 0; i < heights.size(); i++) {
            for (int j = 0; j < heights[0].size(); j++) {
                if (v1[i][j] && v2[i][j]) ans.push_back({i, j});
            }
        }
        return ans;
    }
    void bfs(vector<vector<int>>& heights, vector<vector<int>>& vis, int i, int j){
       queue<pair<int,int>> q;
       int n = heights.size(), m = heights[0].size();
       q.push({i, j});
       vis[i][j] = true;
       while (!q.empty()) {
           pair<int, int> t = q.front(); q.pop();
           int i = t.first, j = t.second;
           for (int k = 0; k < 4; k++) {
               int fx = i + d[k];
               int fy = j + d[k + 1];
               if (fx >= 0 && fy >= 0 && fx < n && fy < m && heights[i][j] <= heights[fx][fy] && !vis[fx][fy]) {
                   q.push({fx, fy});
                   vis[fx][fy] = true;
               } 
           }
       }
    }
};