程序员面试金典 - 面试题 16.19. 水域大小_你有一个用于表示一片土地的整数矩阵land,该矩阵中每个点的值代表对应地点的海拔-CSDN博客

本文链接：https://blog.csdn.net/zjulyx1993/article/details/123293017

题目难度: 中等

原题链接

今天继续更新程序员面试金典系列, 大家在公众号 算法精选 里回复 面试金典 就能看到该系列当前连载的所有文章了, 记得关注哦~

题目描述

你有一个用于表示一片土地的整数矩阵 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

题目思考

如何得到哪些水域相互连通?

解决方案

思路

分析题目, 不难发现这道题的核心在于求连通分量, 唯一不同点在于相邻关系从原来的四方向改为了八方向 (增加了对角连接)
所以我们可以采用经典的 BFS 算法: 遍历所有格子, 如果当前格子是水域, 从它开始向八个方向扩散, 找相邻的水域, 并将访问过的水域格子都标记为已访问, 直到这些点都处理完毕, 对应遍历的格子数目即为当前池塘大小
相比传统的 BFS, 我们这里可以做出一些改进, 从而做到不需要使用额外的 visit 集合来判断哪些点已经被遍历过
具体做法是: 访问过某个水域格子后, 就将其值改为 1, 说白了就是将原有池塘填上, 这样就不会再次处理它
最终得到所有池塘的大小后, 我们只需要将其按照从小到大排序, 即为最终结果
下面的代码对必要步骤有详细的解释, 方便大家理解

复杂度

时间复杂度 O(RC): R 和 C 是矩阵的行和列数, 每个格子只需要访问常数次 (判断或填充)
空间复杂度 O(1): 只使用了常数空间的变量

代码

class Solution:
    def pondSizes(self, land: List[List[int]]) -> List[int]:
        # BFS+8方向+原地更改
        rows, cols = len(land), len(land[0])
        res = []
        for sr in range(rows):
            for sc in range(cols):
                if land[sr][sc] == 0:
                    # 找到一个尚未遍历过的水域, 开始BFS
                    q = [(sr, sc)]
                    # 将原有池塘填上, 更改其值为1, 避免重复遍历
                    land[sr][sc] = 1
                    for r, c in q:
                        for (rr, cc) in ((r - 1, c - 1), (r - 1, c), (r - 1, c + 1), (r, c - 1), (r, c + 1), (r + 1, c - 1), (r + 1, c), (r + 1, c + 1)):
                            # 判断8个方向的相邻格子
                            if 0 <= rr < rows and 0 <= cc < cols and land[rr][cc] == 0:
                                # 当前相邻格子也是水域, 属于连通池塘的一部分
                                q.append((rr, cc))
                                # 同样将其值改为1, 避免重复遍历
                                land[rr][cc] = 1
                    # 最后队列q的长度即为当前池塘大小
                    res.append(len(q))
        # 注意最终需要按照池塘大小排序
        return sorted(res)