【每日一题Day245】面试题 16.19. 水域大小

给定一个表示海拔高度的矩阵，值为0的位置代表水域。程序使用深度优先搜索（DFS）遍历并计算连通的水域单元格，存储每个池塘的大小，并最终返回从小到大排序的池塘大小列表。时间复杂度和空间复杂度均为O(m*n)，其中m和n分别是矩阵的行数和列数。

面试题 16.19. 水域大小

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

dfs感染~

思路

对于每一个水域dfs搜索其八个方向连通的水域大小，添加至动态数组中，最后将动态数组排序后返回
- 为了避免重复访问，访问过一个水域后，将其设置为土地

实现

class Solution {
    int m, n;
    int[][] land;
    public int[] pondSizes(int[][] land) {
        this.m = land.length;
        this.n = land[0].length;
        this.land = land;
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < m; i++){
            for (int j = 0; j < n; j++){
                if (land[i][j] == 0){
                    list.add(dfs(i, j));
                }
            }
        }
        return list.stream().sorted().mapToInt(x -> x).toArray();
    }
    public int dfs(int x, int y){
        int res = 0;
        if (land[x][y] == 0){
            land[x][y] = 1;
            res += 1;
            for (int i = -1; i <= 1; i++){
                for (int j = -1; j <= 1; j++){
                    if (i == 0 && j == 0) continue;
                    int newX = x + i;
                    int newY = y + j;
                    if (newX >= 0 && newX < m && newY >= 0 && newY < n){
                        res += dfs(newX, newY);
                    }
                }
            }
        }
        return res;
    }
}