给定一个由 '1'
(陆地)和 '0'
(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
示例 2:
输入:
11000
11000
00100
00011
输出: 3
解题思路
经典的dfs
问题,通过dfs
遍历所有为'1'
的区域将其标记为'0'
,记录总共遍历的多少次即可。
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
if not grid:
return 0
res = 0
r, c = len(grid), len(grid[0])
d = [0, 1, 0, -1, 0]
def dfs(x, y):
for i in range(4):
nx, ny = x + d[i], y + d[i + 1]
if 0 <= nx < r and 0 <= ny < c and grid[nx][ny] == '1':
grid[nx][ny] = '0'
dfs(nx, ny)
for i, row in enumerate(grid):
for j, _ in enumerate(row):
if grid[i][j] == '1':
res += 1
grid[i][j] = '0'
dfs(i, j)
return res
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!