给你一个由 ‘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
class Solution:
def numIslands(self,grid):
hang=len(grid)
lie=len(grid[0])
visit=set()
ans=0
def dfs(i,j):
for (dx,dy) in [(i,j-1),(i,j+1),(i-1,j),(i+1,j)]:
if dx>=0 and dy>=0 and dx<hang and dy<lie and grid[dx][dy]=="1"\
and (dx,dy) not in visit:
visit.add((dx,dy))
dfs(dx,dy)
for i in range(hang):
for j in range(lie):
if grid[i][j]=="1" and ((i,j) not in visit):
dfs(i,j)
ans+=1
return ans
该问题是一个典型的深度遍历问题,一般来说,深度遍历会存在以下几个步骤:
- 设置一个for循环迭代所有要遍历的结点
- 设置判断条件,当结点符合条件时,进入dfs遍历
- dfs函数会做一次结点的深度遍历,需要注意不同问题返回不同值
- 设置一个集合,每当访问过一个结点,就加入集合中,避免重复访问结点