管科新生首刷leetcode,本题思路来源leetcode官解,写此帖初衷是为了帮助自己更好的理解和“背诵”代码,方便后期复习,望各位大佬多多指教!
题目描述
给你一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
解题思路
对二维表格进行遍历,当遇到值“1”时,将该位置的坐标(r,c)加入待搜索的队列,并将该点的值赋为“0”,同时使岛屿数量加一;对待搜索队列进行出队操作,找到待搜索的节点,对其上下左右((r-1,c),(r+1,c),(r,c-1),(r,c+1))四个位置的节点进行条件判断(是否为“1”,范围是否处在二维表格的范围),若符合判断条件将该点坐标加入队列,并将该点值赋为0,当队列元素遍历完成时,继续对二维表格的元素进行搜索,重复上述操作,直到遍历完二维表格。
python代码实现
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
rn = len(grid)
if rn == 0:
return 0
cn = len(grid[0])
island_nums = 0
# 遍历二维数组
for r in range(rn):
for c in range(cn):
# 遇到岛,island_nums+1,对其上下左右进行一个遍历
#找到最大的一整个岛
if grid[r][c] == "1":
grid[r][c] == "0"
island_nums += 1
neighbors = collections.deque([(r,c)])
while neighbors:
row,col = neighbors.popleft()
for x,y in [(row-1,col),(row+1,col),(row,col-1),(row,col+1)]:
if 0 <= x <rn and 0 <= y < cn and grid[x][y] == "1":
grid[x][y] = "0"
neighbors.append([x,y])
return island_nums