200.岛屿数量- 力扣(LeetCode)

原创于 2025-04-12 13:13:27 发布 · 307 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #python

题目：

给你一个由 '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"],

["0","0","1","0","0"],

["0","0","0","1","1"] ]

输出：3

提示：

m == grid.length
n == grid[i].length
1 <= m, n <= 300
gridi 的值为 '0' 或 '1'

思路如下：

对矩阵进行依次遍历，如果当前grid[i] [j] == "1",则启动DFS模式。找到与之相连的所有1，将其置为0。搜索结束后，我们找到了一个岛屿，岛屿数量+=1。如此循环，最终返回岛屿数量即可。

题解如下：

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        """
        :type grid: List[List[str]]
        :rtype: int
        """
        m, n = len(grid), len(grid[0])

        def dfs(i: int, j: int) -> None:
            """
            :type i: int, j: int
            :rtype: None
            """
            # 出界，或者不是 '1'，就不再往下递归
            if i < 0 or i >= m or j < 0 or j >= n or grid[i][j] != '1':
                return
            grid[i][j] = '0'  # 置0！避免来回横跳无限递归
            dfs(i, j - 1)      # 左
            dfs(i, j + 1)      # 右
            dfs(i - 1, j)      # 上
            dfs(i + 1, j)      # 下

        ans = 0
        for i, row in enumerate(grid):
            for j, c in enumerate(row):
                if c == '1':    # 发现新岛屿
                    dfs(i, j)   # 把这个岛变成水，这样后面遍历到的 '1' 一定是新的岛
                    ans += 1    # 计数+1
        return ans