leetcode-python3算法-岛屿个数

1.题目描述
给定一个由 ‘1’（陆地）和 ‘0’（水）组成的的二维网格，计算岛屿的数量。一个岛被水包围，并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:

输入:
11110
11010
11000
00000

输出: 1
示例 2:

输入:
11000
11000
00100
00011

输出: 3

2.思路
深度优先遍历，遍历矩阵中的每一个元素，如果为1则计数加1，同时把自己以及周围(上下左右)的元素都置0，利用递归实现。

3.代码

• 第一种遍历超时了！

class Solution(object):
    def numIslands(self, grid):
        """
        :type grid: List[List[str]]
        :rtype: int
        """
        if not grid:
            return None
        m = len(grid)
        n = len(grid[0])
        res = 0
        for i in range(m):
            for j in range(n):
                if grid[i][j]:
                    res += 1
                    self.intozero(grid,i,j)
        return res
    
    def intozero(self, grid,i,j):
        if i < 0 or j < 0 or i >= len(grid) or j >= len(grid[0]):
            return 
        if grid[i][j]:
            grid[i][j] = 0
            self.intozero(grid, i + 1, j)
            self.intozero(grid, i - 1, j)
            self.intozero(grid, i, j - 1)
            self.intozero(grid, i, j + 1)

• 后面参考别人的递归部分，将情况细分后OK！

class Solution(object):
    def numIslands(self, grid):
        """
        :type grid: List[List[str]]
        :rtype: int
        """
        if not grid:
            return None
        m = len(grid)
        if m == 0:
            return 0
        n = len(grid[0])
        if n == 0:
            return 0
        res = 0
        for i in range(m):
            for j in range(n):
                if grid[i][j]=='1':
                    res += 1
                    self.intozero(grid,i,j)
        return res
    
    def intozero(self, grid, i, j):
        grid[i][j] = '0'
        # 判断上方字符
        if i > 0 and grid[i - 1][j] == '1':
            self.intozero(grid, i - 1, j)
        # 判断左方字符
        if j > 0 and grid[i][j - 1] == '1':
            self.intozero(grid, i, j - 1)
        # 判断下方字符
        if i < len(grid) - 1 and grid[i + 1][j] == '1':
            self.intozero(grid, i + 1, j)
        # 判断右方字符
        if j < len(grid[0]) - 1 and grid[i][j + 1] == '1':
            self.intozero(grid, i, j + 1)

如果觉得对你有帮助的话，右上角，赞一波。谢谢支持！