LeetCode-热题100：200. 岛屿数量

题目描述

给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

示例 1：

输入： grid = [
[“1”,“1”,“1”,“1”,“0”],
[“1”,“1”,“0”,“1”,“0”],
[“1”,“1”,“0”,“0”,“0”],
[“0”,“0”,“0”,“0”,“0”]
]
输出： 1

示例 2：

输入： grid = [
[“1”,“1”,“0”,“0”,“0”],
[“1”,“1”,“0”,“0”,“0”],
[“0”,“0”,“1”,“0”,“0”],
[“0”,“0”,“0”,“1”,“1”]
]
输出： 3

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 300
• grid[i][j] 的值为’0’ 或 ‘1’

代码及注释

func numIslands(grid [][]byte) int {
    res := 0  // 初始化岛屿数量
    for i := 0; i < len(grid); i++ {  // 遍历行
        for j := 0; j < len(grid[0]); j++ {  // 遍历列
            if grid[i][j] == '1' {  // 如果当前位置是陆地
                dfs(grid, i, j)  // 对该位置进行深度优先搜索
                res++  // 增加岛屿数量
            }
        }
    }
    return res  // 返回岛屿数量
}

func dfs(grid [][]byte, i, j int) {
    if i < 0 || j < 0 || i >= len(grid) || j >= len(grid[0]) {  // 如果越界
        return 
    }
    if grid[i][j] == '0' {  // 如果当前位置是水域
        return
    }
    grid[i][j] = '0'  // 将当前位置标记为水域，表示已访问过
    dfs(grid, i + 1, j)  // 向下搜索
    dfs(grid, i - 1, j)  // 向上搜索
    dfs(grid, i, j + 1)  // 向右搜索
    dfs(grid, i, j - 1)  // 向左搜索
}

代码解释

func numIslands(grid [][]byte) int {
    res := 0
    for i := 0; i < len(grid); i++ {
        for j := 0; j < len(grid[0]); j++ {
            if grid[i][j] == '1' {
                dfs(grid, i, j)
                res++
            }
        }
    }
    return res
}

• numIslands 函数遍历整个二维网格，对于每一个值为 ‘1’ 的格子，调用 dfs 函数进行深度优先搜索，并增加岛屿数量。

func dfs(grid [][]byte, i, j int) {
    if i < 0 || j < 0 || i >= len(grid) || j >= len(grid[0]) {
        return 
    }
    if grid[i][j] == '0' {
        return
    }
    grid[i][j] = '0'
    dfs(grid, i + 1, j)
    dfs(grid, i - 1, j)
    dfs(grid, i, j + 1)
    dfs(grid, i, j - 1)
}

• dfs 函数进行深度优先搜索。
• 首先检查坐标是否越界或者当前格子的值是否为 ‘0’。
• 如果满足条件，将当前格子的值设置为 ‘0’。
• 然后递归地对当前格子的上、下、左、右四个方向进行深度优先搜索。