当前位置:   article > 正文

LeetCode 200.岛屿数量——Python实现(递归、广度优先)_岛屿数量python

岛屿数量python

1、通过递归的方式实现岛屿数量的计算

   (1)寻找某一个 1 为入口,迭代将与它相连的 1 都标记为0,这样直到相连的一整块周围都没有 1 ,只有0的时候,迭代结束 

   (2)完成(1)中的 一个完整的迭代一次,就代表找到了一块岛屿

   (3)在整个网格上循环找为 1 的入口,然后计算岛屿的数量

  1. class Solution:
  2. def numIslands(self, grid):
  3. def mark_zeros(grid, i, j, num_row, num_column):
  4. grid[i][j] = "0"
  5. for x, y in [(i - 1, j), (i + 1, j), (i, j - 1), (i, j + 1)]:
  6. if 0 <= x < num_row and 0 <= y < num_column and grid[x][y] == "1":
  7. mark_zeros(grid, x, y, num_row, num_column)
  8. num_row = len(grid)
  9. if num_row == 0:
  10. return 0
  11. num_column = len(grid[0])
  12. # 定义岛屿的数量
  13. num_island = 0
  14. # 循环找岛屿数
  15. for i in range(num_row):
  16. for j in range(num_column):
  17. if grid[i][j] == "1":
  18. num_island += 1
  19. # 将与该“1”相连的所有“1”标记为“0”,直到无法标记
  20. mark_zeros(grid, i, j, num_row, num_column)
  21. return num_island

2、通过 bfs 的方式实现岛屿数量的查询,跟递归方式的思想和流程一样,只是将每次相连的 "1" 通过迭代的方式标记,变为 将相连的节点加入队列,直到当前队列满员,并且都标记为“0”为止。

  1. class Solution:
  2. def numIslands(self, grid):
  3. import collections
  4. num_row = len(grid)
  5. if num_row == 0:
  6. return 0
  7. num_column = len(grid[0])
  8. # 定义岛屿的数量
  9. num_island = 0
  10. # 循环找岛屿数
  11. for i in range(num_row):
  12. for j in range(num_column):
  13. if grid[i][j] == "1":
  14. num_island += 1
  15. grid[i][j] = "0"
  16. # 将与该“1”相连的所有“1”标记为“0”,直到无法标记
  17. queue = collections.deque([(i, j)])
  18. while queue:
  19. current_x, current_y = queue.popleft()
  20. for x, y in [(current_x - 1, current_y), (current_x + 1, current_y), (current_x, current_y - 1),
  21. (current_x, current_y + 1)]:
  22. if 0 <= x < num_row and 0 <= y < num_column and grid[x][y] == "1":
  23. queue.append((x, y))
  24. grid[x][y] = "0"
  25. return num_island

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/209776
推荐阅读
相关标签
  

闽ICP备14008679号