一、题目
给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false 。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例 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
示例 3:
输入:board = [[“A”,“B”,“C”,“E”],[“S”,“F”,“C”,“S”],[“A”,“D”,“E”,“E”]], word = “ABCB”
输出:false
提示:
m == board.length
n = board[i].length
1 <= m, n <= 6
1 <= word.length <= 15
board 和 word 仅由大小写英文字母组成
进阶:你可以使用搜索剪枝的技术来优化解决方案,使其在 board 更大的情况下可以更快解决问题?
二、解法
深度优先搜索,对于每个点,往下搜就好了。找到符合要求的,就返回true
,如果到最后还找不到,就返回false
就行了
完整代码
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
m, n = len(board), len(board[0])
def dfs(i, x, y):
if i == len(word):
return True
if x < 0 or y < 0 or x >= m or y >= n or board[x][y] != word[i]:
return False
board[x][y] = ''
res = dfs(i + 1, x - 1, y) or dfs(i + 1, x + 1, y) or dfs(i + 1, x, y - 1) or dfs(i + 1, x, y + 1)
board[x][y] = word[i]
return res
for i in range(m):
for j in range(n):
if dfs(0, i, j):
return True
return False