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

最新推荐文章于 2022-04-27 19:18:35 发布

virgilshi

最新推荐文章于 2022-04-27 19:18:35 发布

阅读量276

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： dfs leetcode

本文链接：https://blog.csdn.net/whutshiliu/article/details/105148524

LeetCode 专栏收录该内容

297 篇文章

订阅专栏

本文介绍了一种算法，用于找到一个给定的非负整数矩阵中，哪些单元格的水流可以同时到达太平洋和大西洋。通过从边界开始的深度优先搜索(DFS)，标记可达两个海洋的单元格，最终返回所有符合条件的坐标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

请找出那些水流既可以流动到“太平洋”，又能流动到“大西洋”的陆地单元的坐标。

提示：

输出坐标的顺序不重要
m 和 n 都小于150

在这里插入图片描述
解题思路： 此题一开始想到的是对每个点进行DFS搜索，看是否能找到两条从该点出发，直达两洋的路径，若存在则该点符合要求。但是估计会TLE。最后参考了网友Grandyang的解法，发现此题的解法需要变换一下思路，从四边开始搜索，这个思路在解被围绕的区域这题的时候出现过，也是从四边开始搜。此题如果转换这个思路做的话，就比较简单了。从四条边往中间搜索，将属于pacific/alantic的点分别标记，某点均被标记则表明它均可达pacific和alantic。

class Solution {
public:
    void dfs(vector<vector<int>>& matrix, int r, int c, int h, vector<vector<bool>> &visited) {
        int rows = matrix.size(), cols = matrix[0].size();
        if (r < 0 || r >= rows || c < 0 || c >= cols || visited[r][c] || matrix[r][c] < h) return;
        visited[r][c]  = true;
        dfs(matrix, r + 1, c, matrix[r][c], visited);
        dfs(matrix, r - 1, c, matrix[r][c], visited);
        dfs(matrix, r, c + 1, matrix[r][c], visited);
        dfs(matrix, r, c - 1, matrix[r][c], visited);
    }
    vector<vector<int>> pacificAtlantic(vector<vector<int>>& matrix) {
        if (matrix.empty() || matrix[0].empty()) return {};
        int rows = matrix.size(), cols = matrix[0].size();
        vector<vector<bool>> pacific(rows, vector<bool>(cols, false)), atlantic(rows, vector<bool>(cols, false));
        for (int i = 0; i < rows; ++i) {
            dfs(matrix, i, 0, 0, pacific);
            dfs(matrix, i, cols - 1, 0, atlantic);
        }
        for (int j = 0; j < cols; ++j) {
            dfs(matrix, 0, j, 0, pacific);
            dfs(matrix, rows - 1, j, 0, atlantic);
        }
        vector<vector<int>> res;
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (atlantic[i][j] && pacific[i][j]) {
                    res.push_back({i,j});
                }
            }
        }
        return res;
    }
};