网格图dfs——面试题 16.19. 水域大小（参考灵神）

✊✊✊🌈大家好！本篇文章将较详细介绍一道经典的网格图dfs的题目面试题 16.19. 水域大小。代码语言为：C++代码😇。

---

394. 面试题 16.19. 水域大小

🔒1、题目：

你有一个用于表示一片土地的整数矩阵land，该矩阵中每个点的值代表对应地点的海拔高度。若值为0则表示水域。由垂直、水平或对角连接的水域为池塘。池塘的大小是指相连接的水域的个数。编写一个方法来计算矩阵中所有池塘的大小，返回值需要从小到大排序。
🌲 示例 1🌲：

输入：
	[
	  [0,2,1,0],
	  [0,1,0,1],
	  [1,1,0,1],
	  [0,1,0,1]
	]
输出： [1,2,4]

❗️ 提示 ❗️ ：

• 0 < len(land) <= 1000
• 0 < len(land[i]) <= 1000

来源：力扣（LeetCode）👈

☀️2、思路：

题目中可以枚举八个方向相邻的格子，如果相邻的格子为0，就移动。可以用一个memo数组记录访问过的格子，例如memo[1][2]=true。

• 遍历网格图，从 0 出发，DFS 访问八方向的 0，并把这些0 标记成「访问过」。代码实现时可以直接把0 修改成1。
• DFS 过程中，统计每个方向访问到的格子个数，累加个数得到池塘大小cnt。
  ​- 遍历所有的网格，最后把答案排序返回。

复杂度分析：
⏳时间复杂度：O(mn)，其中 m和 n 分别为 grid 的行数和列数。
🏠空间复杂度：O(mn)。递归最坏需要 O(mn)的栈空间

🔑3、代码：

class Solution {
    int dfs(vector<vector<int>> &land, int x, int y) {
        if (x < 0 || x >= land.size() || y < 0 || y >= land[x].size() || land[x][y])
            return 0;
        land[x][y] = 1; // 标记 (x,y) 被访问，避免重复访问
        int cnt0 = 1;
        for (int i = x - 1; i <= x + 1; i++)
            for (int j = y - 1; j <= y + 1; j++)
                cnt0 += dfs(land, i, j);
        return cnt0;
    }

public:
    vector<int> pondSizes(vector<vector<int>> &land) {
        int m = land.size(), n = land[0].size();
        vector<int> ans;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (land[i][j] == 0) // 从没有访问过的 0 出发
                    ans.push_back(dfs(land, i, j));
        sort(ans.begin(), ans.end());
        return ans;
    }
};