题解 | #岛屿数量#
岛屿数量
https://www.nowcoder.com/practice/0c9664d1554e466aa107d899418e814e
# # 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 # # 判断岛屿数量 # @param grid char字符型二维数组 # @return int整型 # class Solution: def solve(self , grid: List[List[str]]) -> int: # write code here if not grid: return None m = len(grid) n = len(grid[0]) def dfs(grid, i, j): grid[i][j] = '0' if i - 1 >= 0 and grid[i-1][j] == '1': dfs(grid, i-1, j) if i + 1 < m and grid[i+1][j] == '1': dfs(grid, i+1, j) if j - 1 >= 0 and grid[i][j-1] == '1': dfs(grid, i, j-1) if j + 1 < n and grid[i][j+1] == '1': dfs(grid, i, j+1) count = 0 for i in range(m): for j in range(n): if grid[i][j] == '1': count += 1 dfs(grid, i, j) return count
解题思路
- 先处理矩阵为空的情况;
- 然后遍历矩阵,当元素值为1的时候,将计数器加一,并将其周围的元素都置为0(跳转到dfs函数)。
- dfs函数实现了将grid[i][j]连通的为1的元素都置为0。
复杂度
- 时间复杂度为O(mn);
- 空间复杂度为O(mn),栈递归的空间复杂度。