问题描述:
给定一个由 '1'(陆地)和 '0'(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水
平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入: 11110 11010 11000 00000 输出: 1
问题分析:
在二维列表中,如果一个元素的上面元素、下面元素、左边元素或者右边的元素为1,那么说明这些元素属于同一块陆地。使用深度优先算法递归陆地位置,标记同一块陆地。当找到一块陆地时,陆地数量+1,同时搜索它四周的位置是否有陆地,如果有则进行标记0,那么在遍历列表时,将不再认为这是一个新的陆地。
算法分析:
使用双层循环对列表进行遍历,时间复杂度为O(m*n)。最坏情况下,进行m*n次dfs递归调用,空间复杂度为O(m*n)。
def dfs(grid, i, j):
if i<0 or j<0 or i>=len(grid) or j>=len(grid[i]) or grid[i][j]=='0':
return
grid[i][j] = '0'
dfs(grid, i-1,j)
dfs(grid, i, j+1)
dfs(grid, i+1, j)
dfs(grid, i, j-1)
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
res = 0
for i in range(len(grid)):
for j in range(len(grid[i])):
if grid[i][j] == '0':
continue
res += 1
dfs(grid, i , j)
return res