一、题目简述
给定一个m*n的非负矩阵代表大陆中每个单元格的高度,太平洋与矩阵的左、上边相邻,大西洋与矩阵的右、下边相邻。
水流只能沿着四个方向进行流动(上、下、左、右),从一个较高的单元格流向较低或等高的单元格。
寻找既能流到大西洋又能流到太平洋的单元格的坐标。
注意:
- 返回单元格坐标的顺序不影响答案的正确性。
- m与n均小于150。
例如:
Given the following 5x5 matrix:
Pacific ~ ~ ~ ~ ~
~ 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 *
* * * * * Atlantic
Return:
[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (positions with parentheses in above matrix).
二、编程思路
本题目可以分为两个部分就行解决,分别找到能够流到大西洋、太平洋单元格的坐标。自然的思路是使用广度优先搜索,例如求能够流到太平洋的单元格坐标时,可以从矩阵的左、上边开始搜索,寻找需要其流经方向上高度高于或等于当前单元格的格子,并将其状态置为可到达。之后从可到达的单元格上做相同搜索即可。
在获得可到达太平洋与大西洋单元格的坐标之后,取其交集就是能够同时达到大西洋和太平洋的单元格坐标。
三、代码实现
class Solution {
public:
vector<pair<int, int> > pacificAtlantic(vector<vector<int> >& matrix) {
vector<pair<int,int> >ans;
int rows=matrix.size();
if(!rows) return ans;
int cols=matrix[0].size();
if(!cols) return ans;
vector<vector<bool> >statusP(rows,vector<bool>(cols,false));
vector<vector<bool> >statusA(rows,vector<bool>(cols,false));
queue<pair<int,int> >qP;//用于DFS能够流到太平洋的点
queue<pair<int,int> >qA;//用于DFS能够流到大西洋的点
//初始化
for(int i=0;i<rows;i++){
statusP[i][0]=true;
statusA[i][cols-1]=true;
qP.push(pair<int,int>(i,0));
qA.push(pair<int,int>(i,cols-1));
}
for(int i=0;i<cols;i++){
statusP[0][i]=true;
statusA[rows-1][i]=true;
qP.push(pair<int,int>(0,i));
qA.push(pair<int,int>(rows-1,i));
}
int dx[4]={-1,1,0,0};
int dy[4]={0,0,-1,1};
//广度优先搜索能够流到太平洋的点
while(!qP.empty()){
pair<int,int> cur=qP.front();
qP.pop();
int cur_x=cur.first;
int cur_y=cur.second;
for(int i=0;i<4;i++){
int nx=cur_x+dx[i];
int ny=cur_y+dy[i];
if(nx>=0 && nx<rows && ny>=0 && ny<cols && !statusP[nx][ny] && matrix[cur_x][cur_y]<=matrix[nx][ny]){
statusP[nx][ny]=true;
qP.push(pair<int,int>(nx,ny));
}
}
}
//广度优先搜索能够流到大西洋的点
while(!qA.empty()){
pair<int,int> cur=qA.front();
qA.pop();
int cur_x=cur.first;
int cur_y=cur.second;
for(int i=0;i<4;i++){
int nx=cur_x+dx[i];
int ny=cur_y+dy[i];
if(nx>=0 && nx<rows && ny>=0 && ny<cols && !statusA[nx][ny] && matrix[cur_x][cur_y]<=matrix[nx][ny]){
statusA[nx][ny]=true;
qA.push(pair<int,int>(nx,ny));
}
}
}
//寻找既能流到太平洋又能流到大西洋的点
for(int i=0;i<rows;i++){
for(int j=0;j<cols;j++){
if(statusP[i][j] && statusA[i][j]){
ans.push_back(pair<int,int>(i,j));
}
}
}
return ans;
}
};
四、参考资料
无