岛屿数量
题目描述:给定一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。可以假设网格的四个边均被水包围。
示例:
输入:
11000
11000
00100
00001
输出: 3
岛屿数量是一道深度优先搜索的题目,它与回溯法之间就是差了一个不满足就回头的步骤。初始化一个用来记录访问过位置的矩阵。首先遍历矩阵,找到为 “1” 且未被标记过的位置,结果加一,然后在此处进行 DFS,把所有与它相连的陆地全部标记,这样这些陆地在之后的遍历中就不会被当作新的岛屿。最后把结果返回即可。
回溯法说起来简单,也比较套路,但是只有自己动手实现才能发现自己的错误理解。这里使用的方向列表的技巧在类似的题目中都能用到,可以记忆一下。代码如下:
class Solution:
direct = [(1,0), (-1,0), (0,1), (0,-1)]
def numIslands(self, grid: List[List[str]]) -> int:
if not grid:
return 0
m, n = len(grid), len(grid[0])
ans = 0
state = [[False for _ in range(n)] for _ in range(m)]
for i in range(m):
for j in range(n):
if not state[i][j] and grid[i][j]=='1':
ans += 1
self.dfs(i, j, state, grid, m, n)
return ans
def dfs(self, r, c, state, grid, m, n):
state[r][c] = True
for d in self.direct:
new_r = r + d[0]
new_c = c + d[1]
if 0<=new_r<m and 0<=new_c<n and \
grid[new_r][new_c]=='1' and not state[new_r][new_c]:
self.dfs(new_r, new_c, state, grid, m, n)
单词搜索
题目描述:给定一个二维网格和一个单词,找出该单词是否存在于网格中。单词必须按照字母的顺序,通过相邻的单元格内的字母构成,其中相邻单元格是指那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
输入:
board =
[
[A, B, C, E],
[S, F, C, S],
[A, D, E, E]
]
word = ABCCED
输出: true
这道题是一道典型的回溯法题目。比岛屿数量难的地方就在于,如果当前搜索到的字母周围的字母没有符合要求的,那么就要把已经标记访问过了的当前字母的位置重新置为没有访问过,也就是回溯。至于寻找单词本身就是一般的深度优先搜索了。代码如下:
class Solution:
direct = [(1,0), (-1,0), (0,1), (0,-1)]
def exist(self, board: List[List[str]], word: str) -> bool:
m = len(board)
if m == 0: return False
n = len(board[0])
state = [[False for _ in range(n)] for _ in range(m)]
for i in range(m):
for j in range(n):
if self.dfs(i, j, 0, board, state, word, m, n):
return True
return False
def dfs(self, r, c, cur, board, state, word, m, n):
if cur == len(word)-1:
return board[r][c] == word[cur]
if board[r][c] == word[cur]:
state[r][c] = True
for d in self.direct:
new_r = r + d[0]
new_c = c + d[1]
if 0<=new_r<m and 0<=new_c<n and \
not state[new_r][new_c] and \
self.dfs(new_r, new_c, cur+1, board, state, word, m, n):
return True
state[r][c] = False
return False