题目链接
给你一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
示例 1:
输入:grid = [
[“1”,“1”,“1”,“1”,“0”],
[“1”,“1”,“0”,“1”,“0”],
[“1”,“1”,“0”,“0”,“0”],
[“0”,“0”,“0”,“0”,“0”]
]
输出:1
示例 2:
输入:grid = [
[“1”,“1”,“0”,“0”,“0”],
[“1”,“1”,“0”,“0”,“0”],
[“0”,“0”,“1”,“0”,“0”],
[“0”,“0”,“0”,“1”,“1”]
]
输出:3
解题思路
深度优先遍历 DFS
- 遍历 grid 。
- 对于每一个字符为 ‘1’ 的元素,遍历其上下左右四个方向,并将该字符置为 0,保证下次不会被重复遍历。
- 如果超出边界,则返回 0。
- 对于 (i, j) 位置的元素来说,递归遍历的位置就是 (i - 1, j)、(i, j - 1)、(i + 1, j)、(i, j + 1) 四个方向。每次遍历到底,统计数记录一次。
- 最终统计出深度优先搜索的次数就是我们要求的岛屿数量。
解题代码
class Solution:
def numIslands(self, grid: [[str]]) -> int:
def dfs(grid, i, j):
n = len(grid)
m = len(grid[0])
if not 0 <= i < n or not 0 <= j < m or grid[i][j] == '0': return 0
grid[i][j] = '0'
dfs(grid, i + 1, j)
dfs(grid, i, j + 1)
dfs(grid, i, j - 1)
dfs(grid, i - 1, j)
count = 0
for i in range(len(grid)):
for j in range(len(grid[0])):
if grid[i][j] == '1':
dfs(grid, i, j)
count += 1
return count
参考资料:datawhalechina