题目
给定一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围,地址。
示例1
输入:
11110
11010
11000
00000
输出: 1
示例2
输入:
11000
11000
00100
00011
输出: 3
意思就是给出的数组中有多少个连通的岛屿,只要通过上下左右能达到的岛屿都加连通,不连通的岛屿之间隔的是海洋。
解题思路
广度优先搜索(BFS)
找到符合条件的点进行广度优先搜索,把遍历过的点做好标记,搜索的时候设置一个四个方向的数组,然后循环该数组进行移动。
directions = [(-1, 0), (0, -1), (1, 0), (0, 1)]
Python实现
from collections import deque
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
directions = [(-1, 0), (0, -1), (1, 0), (0, 1)]
m = len(grid)
if m == 0:
return 0
n = len(grid[0])
# 设置标志位记录该位置是否遍历过
marked = [[False for _ in range(n)] for _ in range(m)]
count = 0
for i in range(m):
for j in range(n):
# 只有当是陆地并且没有被走过的时候才进入计数
if grid[i][j] == '1' and marked[i][j] == False:
dp = deque()
dp.append((i, j))
marked[i][j] = True
# 开始进行搜索
while dp:
cur_x, cur_y = dp.popleft()
for direction in directions:
new_x, new_y = cur_x + direction[0], cur_y + direction[1]
# 当是陆地,没有被遍历过并且没有超出边界的情况下,放入队列并更新标记位
if 0 <= new_x < m and 0 <= new_y < n and grid[new_x][new_y] == '1' and marked[new_x][new_y] == False:
dp.append((new_x, new_y))
marked[new_x][new_y] = True
# 记录有多少个连通的岛屿
count += 1
return count