给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左
边界和上边界,而“大西洋”处于大陆的右边界和下边界。
规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
题解
深度优先搜索,可以反过来进行思考,从大洋向内地流。
代码
class Solution {
public:
void dfs(vector<vector<int>>& matrix,int x,int y,int m,int n,vector<vector<bool>>& f){
f[x][y]=true;
for (int k=0;k<4;k++){
int xx=x+d[k][0];
int yy=y+d[k][1];
if (xx<0||yy<0||xx>=m||yy>=n) continue;
if (!f[xx][yy]&&matrix[xx][yy]>=matrix[x][y]){
dfs(matrix,xx,yy,m,n,f);
}
}
}
vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
int m=matrix.size();
if (!m) return {};
int n=matrix[0].size();
vector<vector<bool>> f1(m,vector<bool>(n,false));
vector<vector<bool>> f2(m,vector<bool>(n,false));
for (int i=0;i<m;i++){
dfs(matrix,i,0,m,n,f1);
dfs(matrix,i,n-1,m,n,f2);
}
for (int j=0;j<n;j++){
dfs(matrix,0,j,m,n,f1);
dfs(matrix,m-1,j,m,n,f2);
}
vector<vector<int>> res;
for (int i=0;i<m;i++){
for (int j=0;j<n;j++){
if (f1[i][j]&&f2[i][j]) res.push_back({i,j});
}
}
return res;
}
int d[4][2]={0,1,0,-1,1,0,-1,0};
};