leetcode 200. 岛屿数量 DFS
给你一个由 '1'(陆地)和 '0'(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
示例 1:
输入:
[
['1','1','1','1','0'],
['1','1','0','1','0'],
['1','1','0','0','0'],
['0','0','0','0','0']
]
输出: 1
示例 2:
输入:
[
['1','1','0','0','0'],
['1','1','0','0','0'],
['0','0','1','0','0'],
['0','0','0','1','1']
]
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。
链接:https://leetcode-cn.com/problems/number-of-islands
一般是用个visited数组来记录访问,其实这题对于数字为0的不会进行操作的,因此对于已经访问的元素, 可以将其置为0即可。
//Java
class Solution { public int numIslands(char[][] grid) { if(grid==null||grid.length==0||grid[0].length==0)return 0; int res=0; for(int row=0;row<grid.length;row++) for(int col=0;col<grid[0].length;col++) if(grid[row][col]=='1'){ dfs(grid,row,col); res++; } return res; } private void dfs(char[][] grid,int row,int col){ if(row<0||row==grid.length||col<0||col==grid[0].length||grid[row][col]!='1')return; grid[row][col]='0'; dfs(grid,row-1,col); dfs(grid,row+1,col); dfs(grid,row,col-1); dfs(grid,row,col+1); } }
//python
class Solution: def numIslands(self, grid: List[List[str]]) -> int: if not grid: return 0 res = 0 for i in range(len(grid)): for j in range(len(grid[0])): if grid[i][j] == '1': self.dfs(grid, i, j) res += 1 return res def dfs(self, grid, i, j): if i < 0 or j < 0 or i >= len(grid) or j >= len(grid[0]) or grid[i][j] != '1': return grid[i][j] = '0' self.dfs(grid, i + 1, j) self.dfs(grid, i - 1, j) self.dfs(grid, i, j + 1) self.dfs(grid, i, j - 1)