[LeetCode] 面试题 16.19. 水域大小

你有一个用于表示一片土地的整数矩阵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/problems/pond-sizes-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

这是属于 flood fill 那一类的题。这里我分享一个 BFS 的做法。唯一需要注意的细节是这道题里池塘的定义是八个方向，以及需要对最后的结果先排序再输出。

时间O(mn)

空间O(mn) - 最坏情况下只有一个池塘，即整个池塘，所以所有的点都会在queue内走一遍

Java实现

class Solution {
    int m;
    int n;
    int[] dx = {-1, -1, -1, 0, 0, 1, 1, 1};
    int[] dy = {1, 0, -1, 1, -1, 1, 0, -1};
    List<Integer> list;

    public int[] pondSizes(int[][] land) {
        m = land.length;
        n = land[0].length;
        list = new ArrayList<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (land[i][j] == 0) {
                    int count = bfs(land, i, j);
                    list.add(count);
                }
            }
        }
        int n = list.size();
        int[] res = new int[n];
        for (int i = 0; i < n; i++) {
            res[i] = list.get(i);
        }
        Arrays.sort(res);
        return res;
    }

    private int bfs(int[][] land, int i, int j) {
        int count = 1;
        land[i][j] = -1;
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(new int[] { i, j });
        while (!queue.isEmpty()) {
            int[] cur = queue.poll();
            int x = cur[0];
            int y = cur[1];
            for (int k = 0; k < 8; k++) {
                int newX = x + dx[k];
                int newY = y + dy[k];
                if (newX >= 0 && newX < m && newY >= 0 && newY < n && land[newX][newY] == 0) {
                    land[newX][newY] = -1;
                    count++;
                    queue.offer(new int[] { newX, newY });
                }
            }
        }
        return count;
    }
}

 

LeetCode 题目总结