给定一个 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;
}
};