面试题 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; } }
复杂度
- 时间复杂度：O ( m ∗ n )
- 空间复杂度：O ( m ∗ n )
