岛屿系列问题
岛屿系列题?的核?考点就是? DFS/BFS 算法遍历?维数组。 那么如何在?维矩阵中使? DFS 搜索呢?如果你把?维矩阵中的每?个位置看做?个节点,这个节点的上下左右四个位置就是相邻节点,那么整个矩阵就可以抽象成?幅?状的「图」结构。 可以根据?叉树的遍历框架改写出?维矩阵的 DFS 代码框架:
def traverse(root):
traverse(root.left)
traverse(root.right)
def dfs(grid, i, j, visited):
m, n = len(grid), len(grid[0])
if i < 0 or j < 0 or i >= m or j >= n:
return
if visited[i][j]:
return
visited[i][j] = 1
dfs(grid, i - 1, j)
dfs(grid, i + 1, j)
dfs(grid, i, j - 1)
dfs(grid, i, j + 1)
解法:DFS
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
m, n = len(grid), len(grid[0])
res = 0
for i in range(m):
for j in range(n):
if grid[i][j] == '1':
res += 1
self.dfs(i, j, grid)
return res
def dfs(self, row, col, grid):
m, n = len(grid), len(grid[0])
if row < 0 or row >= m or col < 0 or col >= n:
return
if grid[row][col] == '0':
return
grid[row][col] = '0'
directions = [(-1, 0), (1, 0), (0, -1), (0, 1)]
for i, j in directions:
self.dfs(row+i, col+j, grid)
为什么每次遇到岛屿,都要? DFS 算法把岛屿「淹了」呢?主要是为了省事,避免维护 visited 数组。 因为 dfs 函数遍历到值为 0 的位置会直接返回,所以只要把经过的位置都设置为 0,就可以起到不?回头路的作?。
解法:模拟 根据题意,明确终止条件,随后进行模拟即可。具体如下: 计算一个整数的各位相加的做法是,每次计算当前整数除以 10 的余数得到最低位数,将最低位数加到总和中,然后将当前整数除以 10。重复上述操作直到当前整数变成 0,此时的总和即为原整数各位相加的结果。
class Solution:
def addDigits(self, num: int) -> int:
while num >= 10:
temp = 0
while num != 0:
temp += num % 10
num //= 10
num = temp
return num
|