2021-06-09

leetcode200-DFS:岛屿数量

---

200. 岛屿数量
     给你一个由 ‘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’

---

解题思路

DFS

代码

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        count=0
        if len(grid)==0:
            return count
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                if grid[i][j]=='1':
                    self.DFS(grid,i,j)
                    count+=1
        return count
    def DFS(self,grid,row,column):
        r=len(grid)
        c=len(grid[0])
        grid[row][column]='0'
        #遍历上下左右
        if row-1>=0 and grid[row-1][column]=='1':
            self.DFS(grid,row-1,column)
        if row+1<r and grid[row+1][column]=='1':
            self.DFS(grid,row+1,column)
        if column-1>=0 and grid[row][column-1]=='1':
            self.DFS(grid,row,column-1)
        if column+1<c and grid[row][column+1]=='1':
            self.DFS(grid,row,column+1)