题目描述
给你一个由 '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’。
- 然后递归地对当前格子的上、下、左、右四个方向进行深度优先搜索。