【算法修炼道路】水域大小

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

示例：

输入：
[
  [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）
链接：https://leetcode-cn.com/problems/pond-sizes-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：递归，搜索，8个方向的深搜

class Solution {
public:
    vector<int> pondSizes(vector<vector<int>>& land) {
        vector<int> ponds;
        // 遍历每一个池塘中的格子
         for (int i=0; i<land.size(); ++i) {
             for (int j=0;j<land[i].size(); ++j) {
                if (land[i][j] == 0) { // 如果发现一个为0的格子，即发现一块水域，则以此为圆心开始进行递归搜索
                    int size = dfs(land, land.size(), land[0].size(), i, j);
                    ponds.push_back(size);
                }
             }
         }
         sort(ponds.begin(), ponds.end());
         return ponds;
    }

    // land为池塘，row为池塘的行数，col为池塘的列数，i j 为当前所处的池塘位置
    int dfs(vector<vector<int>> & land,int row, int col, int i, int j) {
        // 边界处理，如果超出了池塘的边界，返回0
        if (i < 0 || j < 0 || i >= row || j >= col) {
            return 0;
        }
        if (land[i][j] == 0) { // 如果当前位置是水域，则以此为圆心，周围8个方向为路径，进行深度优先搜索
            land[i][j] = 1; // 每经过一块区域，需要对此处进行标记，表示已经来过，避免重复计算
            // 返回池塘的大小，大小为当前的1 加上其他八个方向的大小
            return 1 + dfs(land, row, col, i-1, j-1) +
                    dfs(land, row, col, i-1, j) +
                    dfs(land, row, col, i-1, j+1) +
                    dfs(land, row, col, i, j-1) +
                    dfs(land, row, col, i, j+1) +
                    dfs(land, row, col, i+1, j-1) +
                    dfs(land, row, col, i+1, j) +
                    dfs(land, row, col, i+1, j+1);
        }
        else {  // 当前位置不是水域，返回0，表示此路不通，停止搜索
            return 0;
        }
    }
};

执行结果：

通过

显示详情

执行用时：264 ms, 在所有 C++ 提交中击败了92.48%的用户

内存消耗：31.6 MB, 在所有 C++ 提交中击败了52.13%的用户

（妹想到运行还挺快的）