Swift刷算法

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
    }
}
image.png
上一篇 下一篇

猜你喜欢

热点阅读