有一个 m × n
的矩形岛屿,与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界,而 “大西洋” 处于大陆的右边界和下边界。
这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n
的整数矩阵 heights
, heights[r][c]
表示坐标 (r, c)
上单元格 高于海平面的高度 。
岛上雨水较多,如果相邻单元格的高度 小于或等于 当前单元格的高度,雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
返回网格坐标 result
的 2D 列表 ,其中 result[i] = [ri, ci]
表示雨水从单元格 (ri, ci)
流动 既可流向太平洋也可流向大西洋 。
示例 1:
输入: 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]]
示例 2:
输入: heights = [[2,1],[1,2]] 输出: [[0,0],[0,1],[1,0],[1,1]]
提示:
m == heights.length
n == heights[r].length
1 <= m, n <= 200
0 <= heights[r][c] <= 105
题目倒不难,从“太平洋”“大西洋”反向搜索可以联通的点(大于等于前一个),找出同时在两个vis数组都出现的值返回即可,但是写了我好久,所以记录一下。。。
class Solution {
public:
int vis1[305][305], vis2[305][305];
vector<vector<int>>ans;
int n, m, dx[4] = { -1,0,1,0 }, dy[4] = { 0,1,0,-1 };
void dfs(int x, int y, vector<vector<int>>& heights, int vis[305][305]) {
if (x < 0 || x >= n || y < 0 || y >= m || vis[x][y]) return;
vis[x][y] = 1;
for (int i = 0;i < 4;i++) {
int xx = x + dx[i];
int yy = y + dy[i];
if (xx >= 0 && xx < n && yy >= 0 && yy < m && heights[xx][yy] >= heights[x][y]) {
dfs(xx, yy, heights, vis);
}
}
}
vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
n = heights.size(), m = heights[0].size();
for (int i = 0;i < m;i++)
dfs(0, i, heights, vis1);
for (int i = 0;i < n;i++)
dfs(i, 0, heights, vis1);
for (int i = 0;i < m;i++)
dfs(n - 1, i, heights, vis2);
for (int i = 0;i < n;i++)
dfs(i, m - 1, heights, vis2);
for (int i = 0;i < n;i++) {
for (int j = 0;j < m;j++) {
if (vis1[i][j] && vis2[i][j]) ans.push_back({ i,j });
}
}
return ans;
}
};