Swift刷算法:岛屿数量
2022-06-19 本文已影响0人
JonorZhang
给你一个由
'1'
(陆地)和'0'
(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
题目来源:https://leetcode.cn/problems/number-of-islands
class Solution {
func numIslands(_ grid: [[Character]]) -> Int {
// 变成可变的才好原地修改
var grid = grid
func dfs(_ i: Int, _ j: Int) {
// 数组四周的上下左右方向可能会越界,要排除掉
if i < 0 || j < 0 || i >= grid.count || j >= grid[i].count {
return
}
// 如果已经是海水就不需要在遍历
if grid[i][j] == "0" {
return
}
// 填充为海水,避免下次再次遍历到出现循环
grid[i][j] = "0"
// 使用深度优先搜素四个方向
dfs(i+1, j) // 下
dfs(i-1, j) // 上
dfs(i, j+1) // 右
dfs(i, j-1) // 左
}
// 岛屿数量
var res = 0
for i in 0 ..< grid.count {
for j in 0 ..< grid[i].count {
// 遇到"1"岛屿数量加1,同时用dfs把该岛屿变成海水
if grid[i][j] == "1" {
res += 1
dfs(i, j)
}
}
}
return res
}
}
![](https://img.haomeiwen.com/i1806503/1864fab7b165da07.png)