有一个 m × n 的矩形岛屿,与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界,而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights , heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
岛上雨水较多,如果相邻单元格的高度 小于或等于 当前单元格的高度,雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回 网格坐标 result 的 2D列表 ,其中 result[i] = [ri, ci] 表示雨水可以从单元格 (ri, ci) 流向 太平洋和大西洋 。
输入: heights = [[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]]
貌似图论有一丢丢入门了,这个题没看解析自己写出来了。
思路
类似多元最短路径。
求出可以到太平洋的所有点,再求出可以到达大西洋的所有点。交集就是可以到达两个洋的点。
先说一下怎么求可以到太平洋的所有点,大西洋类似。
初始化一个105*105的数组ArrP,初始化为0
已知,太平洋附近的所有点都可以到太平洋,即上边和左边的所有元素,将他们压入队列,并置arrP为1,表示可以到海洋,然后再遍历这些点,找到和它相同或比它高并且arrP为0的岛屿,将其设为1(防止重复查找)并压入队列。当队列为空时,已经找到全部可以流入太平洋的岛屿。
类似找大西洋,arrP和arrA都为1的点就是我们要的答案
class Solution {
public:
//初始化数组用于存储
int arrP[205][205]={0};
int arrA[205][205]={0};
//四个方向
int x[4]={1,-1,0,0};
int y[4]={0,0,1,-1};
//查找可入洋的岛屿,数组对应位置置为1
void get(vector<vector<int>>& heights,int (&arr)[205][205] )
{
int m=heights.size();
int n=heights[0].size();
//初始的点压入队列
queue<pair<int,int >> qu;
for(int i=0;i<m;++i)
{
for(int j=0;j<n;++j)
{
if(arr[i][j]==1)
{
qu.push({i,j});
}
}
}
while(!qu.empty())
{
auto q=qu.front();
qu.pop();
int i=q.first,j=q.second;
for(int k=0;k<4;k++)
{
int vi=i+x[k];
int vj=j+y[k];
//坐标合法并且未被访问过,且高度高于或等于的岛屿
if(vi>=0&&vi<m&&vj>=0&&vj<n&&heights[vi][vj]>=heights[i][j]&&!arr[vi][vj])
{
qu.push({vi,vj});
arr[vi][vj]=1;
}
}
}
}
vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
int m=heights.size();
int n=heights[0].size();
//先标记已知必定可以入洋的点
for(int i=0;i<m;++i)
{
arrP[i][0]=1;
arrA[i][n-1]=1;
}
for(int i=0;i<n;i++)
{
arrP[0][i]=1;
arrA[m-1][i]=1;
}
vector<vector<int>> ans;
get(heights,arrP);
get(heights,arrA);
//都可以入洋的点计入结果中
for(int i=0;i<m;++i)
{
for(int j=0;j<n;++j)
{
if(arrA[i][j]&&arrP[i][j])
{
ans.push_back({i,j});
}
}
}
return ans;
}
};