- 标签:数组、回溯、矩阵
- 难度:中等
描述:给定一个
要求:如果 True
,否则返回 False
。
说明:
- 单词必须按照字母顺序通过上下左右相邻的单元格字母构成。且同一个单元格内的字母不允许被重复使用。
-
$m == board.length$ 。 -
$n == board[i].length$ 。 -
$1 \le m, n \le 6$ 。 -
$1 \le word.length \le 15$ 。 -
$board$ 和$word$ 仅由大小写英文字母组成。
示例:
- 示例 1:
输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出:true
- 示例 2:
输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "SEE"
输出:true
使用回溯算法在二维矩阵
设函数 backtrack(i, j, index)
表示从 True
,否则返回 False
。
backtrack(i, j, index)
执行步骤如下:
- 如果
$board[i][j] = word[index]$ ,而且 index 已经到达 word 字符串末尾,则返回 True。 - 如果
$board[i][j] = word[index]$ ,而且 index 未到达 word 字符串末尾,则遍历当前位置的所有相邻位置。如果从某个相邻位置能搜索到后缀子串,则返回 True,否则返回 False。 - 如果
$board[i][j] \ne word[index]$ ,则当前字符不匹配,返回 False。
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
directs = [(0, 1), (0, -1), (1, 0), (-1, 0)]
rows = len(board)
if rows == 0:
return False
cols = len(board[0])
visited = [[False for _ in range(cols)] for _ in range(rows)]
def backtrack(i, j, index):
if index == len(word) - 1:
return board[i][j] == word[index]
if board[i][j] == word[index]:
visited[i][j] = True
for direct in directs:
new_i = i + direct[0]
new_j = j + direct[1]
if 0 <= new_i < rows and 0 <= new_j < cols and visited[new_i][new_j] == False:
if backtrack(new_i, new_j, index + 1):
return True
visited[i][j] = False
return False
for i in range(rows):
for j in range(cols):
if backtrack(i, j, 0):
return True
return False
-
时间复杂度:$O(m \times n \times 2^l)$,其中
$m$ 、$n$ 为二维矩阵$board$ 的行数和列数。$l$ 为字符串$word$ 的长度。 - 空间复杂度:$O(m \times n)$。