今天这个题我是题目都看了老半天才给看明白,要找到能够同时流向太平洋和大西洋的区域,然后我也不知道脑子怎么想的,我就觉得只有左下角和右上角的点能够同时流向太平洋和大西洋,然后我就想那只要找到能够流向这两个点的区域就好了,那我就只需要对这两个点深度优先或者广度优先按照“水往高处流”搜索即可找到所有点,为了防止重复利用一个数组记录搜索过的点即可,写完过后一对答案傻了,如果能够同时通向左上方边缘的点和右下方边缘的点也是可以的呀!不过还好,修改起来倒也不算太难,要用俩数组分别记录到达太平洋和大西洋的点,还有就是要遍历的范围变成了第一行第一列最后一行最后一列,代码如下
class Solution
{
vector<vector<int>> ans;
vector<vector<int>> dirs = { {1, 0}, {0, 1}, {-1, 0}, {0, -1} };
vector<vector<int>> getAroundPoint(int x, int y, int row, int col)
{
//获取周围的点
vector<vector<int>> res;
for (auto dir : dirs)
{
if ((x + dir[0] >= 0 && x + dir[0] < row) && (y + dir[1] >= 0 && y + dir[1] < col))
res.push_back({ x + dir[0] , y + dir[1] });
}
return res;
}
void DFS(vector<vector<int>>& heights, vector<vector<bool>> &isOK, int x, int y, int row, int col)
{
//深度优先遍历
if (isOK[x][y])
return;
isOK[x][y] = true;
vector<vector<int>> AroundPoint = getAroundPoint(x, y, row, col);
for (auto point : AroundPoint)
{
if (heights[x][y] <= heights[point[0]][point[1]])
DFS(heights, isOK, point[0], point[1], row, col);
}
}
public:
vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights)
{
//获取小岛长度和宽度
int row = heights.size(), col = heights[0].size();
//建立数组记录哪些区域能够到达太平洋和大西洋
vector<vector<bool>> atlantic(row, vector<bool>(col, false));
vector<vector<bool>> pacific(row, vector<bool>(col, false));
//遍历找到能够到达太平洋和大西洋的点
for (int i = 0; i < col; ++i)
{
DFS(heights, pacific, 0, i, row, col);
DFS(heights, atlantic, row - 1, i, row, col);
}
for (int i = 0; i < row; ++i)
{
DFS(heights, pacific, i, 0, row, col);
DFS(heights, atlantic, i, col - 1, row, col);
}
for (int i = 0; i < row; ++i)
{
for (int j = 0; j < col; ++j)
{
if (pacific[i][j] && atlantic[i][j])
ans.push_back({ i, j });
}
}
return ans;
}
};