跟着专注于计算机视觉的AndyJ的妈妈我学算法之每日一题leetcode200岛屿数量

岛屿数量，用BFS/DFS。还看有用并查集的。
用DFS就好了。
好了，看题：

200. 岛屿数量
给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。
岛屿总是被水包围，并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。
此外，你可以假设该网格的四条边均被水包围。

示例 1:
输入:
[
['1','1','1','1','0'],
['1','1','0','1','0'],
['1','1','0','0','0'],
['0','0','0','0','0']
]
输出: 1

示例 2:
输入:
[
['1','1','0','0','0'],
['1','1','0','0','0'],
['0','0','1','0','0'],
['0','0','0','1','1']
]
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。

code：

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        if not grid: return 0
        self.grid = grid
        self.m = len(grid)
        self.n = len(grid[0])
        self.direct = [[0,1],[1,0],[-1,0],[0,-1]]
        self.visited = {} # 用list会超时间；或者可以和答案一样，把连通的改0
        count = 0
        for i in range(self.m):
            for j in range(self.n):
                if grid[i][j] == '1':
                    if (i,j) in self.visited:
                        continue
                    count += 1
                    self.dfs(i, j)
        return count
    
    def dfs(self, i, j):
        self.visited[(i,j)] = 1
        for di, dj in self.direct:
            tempi = i + di
            tempj = j + dj
            if tempi < 0 or tempj < 0 or tempi >=self.m or tempj >= self.n:
                continue
            if (tempi, tempj) in self.visited:
                continue
            if self.grid[tempi][tempj] == '1':
                self.dfs(tempi, tempj)

好了。BFS自己写。