417.太平洋大西洋水流问题

最新推荐文章于 2024-05-06 19:08:06 发布

High0.0

最新推荐文章于 2024-05-06 19:08:06 发布

阅读量130

点赞数 1

分类专栏： Leetcode 文章标签： leetcode 深度优先搜索 c++

本文链接：https://blog.csdn.net/Linke66/article/details/121069070

版权

Leetcode 专栏收录该内容

56 篇文章 3 订阅

订阅专栏

在这里插入图片描述
解题思路：

一开始自己的思路是看每一个点是否能流到太平洋和大西洋，很复杂，各种细节考虑不到。看了题解觉得这个方法很好，记录一下。

我们从四个边开始往高处走，能走到的位置就是能流到对应边旁边的洋流。

can_reach_p，（太平洋 pacific）代表能到达太平洋；
can_reach_a，（大西洋 atlantic）代表能到达大西洋。

假如某个点的can_reach_p和can_reach_a都为true，说明符合条件。

具体思路看代码。

class Solution {
public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        if(heights.empty()||heights[0].empty()) return {};
        int m=heights.size(),n=heights[0].size();
        vector<vector<int>>ret;
        vector<vector<bool>> can_reach_p(m,vector<bool>(n,false)); //能到达太平洋 pacific ocean
        vector<vector<bool>> can_reach_a=can_reach_p;              //能到达大西洋

        for(int i=0;i<m;++i){
            dfs(heights,can_reach_p,i,0);
            dfs(heights,can_reach_a,i,n-1);
        }

        for(int i=0;i<n;++i){
            dfs(heights,can_reach_p,0,i);
            dfs(heights,can_reach_a,m-1,i);
        }

        for(int i=0;i<m;++i){
            for(int j=0;j<n;++j){
                if(can_reach_a[i][j]&&can_reach_p[i][j]){
                    ret.push_back({i,j});
                }
            }
        }
        return ret; 
    }

    void dfs(const vector<vector<int>>& heights,
                vector<vector<bool>> &can_reach,
                int x,int y)
    {
        if(can_reach[x][y]==true) return;
        can_reach[x][y]=true;

        if(x-1>=0&&heights[x-1][y]>=heights[x][y]) 
            dfs(heights,can_reach,x-1,y);
        if(y-1>=0&&heights[x][y-1]>=heights[x][y]) 
            dfs(heights,can_reach,x,y-1);
        if(x+1<heights.size()&&heights[x+1][y]>=heights[x][y]) 
            dfs(heights,can_reach,x+1,y);
        if(y+1<heights[0].size()&&heights[x][y+1]>=heights[x][y]) 
            dfs(heights,can_reach,x,y+1);
    }
};