leetcode 200. 岛屿数量 Python

题目

给定一个由 ‘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