LeetCode Python島の数
1159 ワード
'1'
(陸地)と'0'
(水)からなる2次元メッシュを与え,島の数を計算した.1つの島は水に囲まれ、水平方向または垂直方向に隣接する陸地で接続されている.グリッドの4つのエッジが水で囲まれていると仮定できます.例1:
:
11110
11010
11000
00000
: 1
class Solution(object):
def numIslands(self, grid):
row = len(grid)
if row == 0 or grid == None:
return 0
col = len(grid[0])
count = 0
for i in range(row):
for j in range(col):
if grid[i][j] == 1:
self.dfsFind(grid, i, j, row, col)
count += 1
return count
def dfsFind(self, grid, i, j, row, col):
if grid[i][j] == 1:
grid[i][j] = 0
if i - 1 >= 0:
grid = self.dfsFind(grid, i - 1, j, row, col)
if i + 1 < row:
grid = self.dfsFind(grid, i + 1, j, row, col)
if j - 1 >= -0:
grid = self.dfsFind(grid, i, j - 1, row, col)
if j + 1 < col:
grid = self.dfsFind(grid, i, j + 1, row, col)
return grid