给定一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-islands
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
——————————
解体思路:遍历整个地图,从标记为1的点进行广度优先遍历,当遇到一个label为1的点时,d+1,d代表的是岛屿的数量,当广度搜索完一片label为1的区域之后,将对应的label变为0。
不断重复上面的操作,一直到遍历完整个地图。
class Solution:
directions = [(-1, 0), (0, -1), (1, 0), (0, 1)]
def numIslands(self, grid: List[List[str]]) -> int:
if not grid:
return 0
rows = len(grid)
cols = len(grid[0])
def neighbor(i,j):
for nr,nc in [[i,j-1],[i,j+1],[i-1,j],[i+1,j]]:
if 0<=nr<rows and 0<=nc<cols:
yield nr,nc
d = 0
for i in range(rows):
for j in range(cols):
if grid[i][j] == '1':
d += 1
grid[i][j]='0'
queue = collections.deque([[i,j]])
while queue:
ii,jj = queue.popleft()
for nr,nc in neighbor(ii,jj):
if grid[nr][nc] == '1':
grid[nr][nc]='0'
queue.append([nr,nc])
return d