0
点赞
收藏
分享

微信扫一扫

LeetCode每日一题刷题记录417.太平洋大西洋水流问题

今天这个题我是题目都看了老半天才给看明白,要找到能够同时流向太平洋和大西洋的区域,然后我也不知道脑子怎么想的,我就觉得只有左下角和右上角的点能够同时流向太平洋和大西洋,然后我就想那只要找到能够流向这两个点的区域就好了,那我就只需要对这两个点深度优先或者广度优先按照“水往高处流”搜索即可找到所有点,为了防止重复利用一个数组记录搜索过的点即可,写完过后一对答案傻了,如果能够同时通向左上方边缘的点和右下方边缘的点也是可以的呀!不过还好,修改起来倒也不算太难,要用俩数组分别记录到达太平洋和大西洋的点,还有就是要遍历的范围变成了第一行第一列最后一行最后一列,代码如下

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;
    }
};
举报

相关推荐

0 条评论