417. 太平洋大西洋水流问题/C++

最新推荐文章于 2023-02-23 18:01:37 发布

原创最新推荐文章于 2023-02-23 18:01:37 发布 · 674 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种基于深度优先搜索(DFS)的算法，用于解决太平洋和大西洋水流交汇问题。通过从每个海洋的边界点开始进行DFS，并标记可以到达的网格点，最后找出同时能被两大洋水流覆盖的网格位置。此算法适用于二维矩阵地形，能够有效识别出受双重海洋影响的区域。

在这里插入图片描述
参考：https://blog.youkuaiyun.com/qq_41855420/article/details/88744535
分别从太平洋的每个点dfs，并记录
分别从大西洋的每个点dfs，并记录
然后遍历所有点，并查看以上2个记录即可。

class Solution {
    vector<vector<int>> res;
    int row,col;
    int d[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
    map<pair<int, int>, bool> pacificFlag;
    map<pair<int, int>, bool> atlanticFlag;
    
    bool inArea(int x,int y){
        return x>=0 && x<row && y>=0 && y<col;
    }
    
    void dfs(vector<vector<int>>& matrix,map<pair<int, int>, bool>& flag, int x,int y){
        flag[{x,y}]=true;
        for(int i=0;i<4;++i){
            int newX=x+d[i][0];
            int newY=y+d[i][1];
            if(inArea(newX,newY) && matrix[x][y]<=matrix[newX][newY] 
            	&& flag.count({newX,newY})==0){
                flag[{newX,newY}]=true;
                dfs(matrix,flag,newX,newY);
            }
        }
        return;
    }
    
   
public:
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
        res.clear();
        row = matrix.size();
        if(row==0)
            return res;
        
        col=matrix[0].size();
        if(col==0)
            return res;
        
        for(int i=0;i<row;++i){
            dfs(matrix,pacificFlag,i,0);
            dfs(matrix,atlanticFlag,i,col-1);
        }
        
        for(int i=0;i<col;++i){
            dfs(matrix,pacificFlag,0,i);
            dfs(matrix,atlanticFlag,row-1,i);
        }
        
        for(int i=0;i<row;++i)
            for(int j=0;j<col;++j)
                if(pacificFlag[{i,j}] && atlanticFlag[{i,j}])
                    res.push_back({i,j});
        return res;
    }
};