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

  1. 题目描述
给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左
边界和上边界,而“大西洋”处于大陆的右边界和下边界。

规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。

请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。
  1. 题解
深度优先搜索,可以反过来进行思考,从大洋向内地流。
  1. 代码
class Solution {
public:
    void dfs(vector<vector<int>>& matrix,int x,int y,int m,int n,vector<vector<bool>>& f){
        f[x][y]=true;
        for (int k=0;k<4;k++){
            int xx=x+d[k][0];
            int yy=y+d[k][1];
            if (xx<0||yy<0||xx>=m||yy>=n) continue;
            if (!f[xx][yy]&&matrix[xx][yy]>=matrix[x][y]){
                dfs(matrix,xx,yy,m,n,f);
            }
        }
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
        int m=matrix.size();
        if (!m) return {};
        int n=matrix[0].size();
        vector<vector<bool>> f1(m,vector<bool>(n,false));
        vector<vector<bool>> f2(m,vector<bool>(n,false));
        for (int i=0;i<m;i++){
            dfs(matrix,i,0,m,n,f1);
            dfs(matrix,i,n-1,m,n,f2);
        }
        for (int j=0;j<n;j++){
            dfs(matrix,0,j,m,n,f1);
            dfs(matrix,m-1,j,m,n,f2);
        }
        vector<vector<int>> res;
        for (int i=0;i<m;i++){
            for (int j=0;j<n;j++){
                if (f1[i][j]&&f2[i][j]) res.push_back({i,j});
            }
        }
        return res;
    }
    int d[4][2]={0,1,0,-1,1,0,-1,0};
};
上一篇:云数据库PostgreSQL新人怎样上手


下一篇:研究生算法 2-1 YY and Lucky Number