【DFS】太平洋大西洋水流问题

最新推荐文章于 2024-11-11 16:07:51 发布

敲击石头的一百零一次

最新推荐文章于 2024-11-11 16:07:51 发布

阅读量263

点赞数 1

分类专栏： ds 文章标签：深度优先算法

本文链接：https://blog.csdn.net/2202_75999061/article/details/140702445

版权

ds 专栏收录该内容

85 篇文章 0 订阅

订阅专栏

一、题目

. - 力扣（LeetCode）. - 备战技术面试？力扣提供海量技术面试资源，帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/pacific-atlantic-water-flow/submissions/549753718/

二、题解

统计两个洋的水分别可以到哪些地方，然后再统计都能到达的地方

class Solution {
public:
    int n, m, g[210][210], gg[210][210], check[210][210], dx[4]{0, 0, 1, -1}, dy[4]{1, -1, 0, 0};

    void dfs(int i, int j, vector<vector<int>>& he, int c) {
        check[i][j] = 1;
        if (c) g[i][j]++; else gg[i][j]++;
        for (int k = 0; k < 4; k++)
            if (i + dx[k] >= 0 && i + dx[k] < n && j + dy[k] >= 0 && j + dy[k] < m && !check[i + dx[k]][j + dy[k]] && he[i][j] <= he[i + dx[k]][j + dy[k]])
                dfs(i + dx[k], j + dy[k], he, c);
    }

    vector<vector<int>> pacificAtlantic(vector<vector<int>>& heights) {
        n = heights.size(), m = heights[0].size();
        for (int i = 0; i < n; i++) 
            if (!check[i][0])
                dfs(i, 0, heights, 0);
        for (int j = 0; j < m; j++)
            if (!check[0][j])
                dfs(0, j, heights, 0);

        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++)
                check[i][j] = 0;

        for (int i = 0; i < n; i++) 
            if (!check[i][m - 1])
                dfs(i, m - 1, heights, 1);
        for (int j = 0; j < m; j++)
            if (!check[n - 1][j])
                dfs(n - 1, j, heights, 1);

        vector<vector<int>> res;
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++)
                if (g[i][j] && gg[i][j])
                    res.push_back({i, j});
       

        return res;
    }
};