题目:
给你一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
解答:
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
#directions:上下左右
directions=[[-1,0],[1,0],[0,-1],[0,1]]
m=len(grid)
n=len(grid[0])
#使用标记避免重复遍历:'2'(已遍历过的陆地格子)
def inArea(r,c):
if 0<=r<m and 0<=c<n:
return True
else:
return False
#将从(r,c)出发组成岛屿的格子置为'2'
def dfs(r,c):
if not inArea(r,c):
return
#若当前格子不是陆地 或者 已经遍历过
if grid[r][c]!='1':
return
#将格子标记为已遍历过
grid[r][c]='2'
#访问当前格子相邻的四个点
for dr,dc in directions:
dfs(r+dr,c+dc)
res=0
for i in range(m):
for j in range(n):
if grid[i][j]=='1':
res+=1
dfs(i,j)
return res